Wrap task calls into AstStmtExpr

Signed-off-by: Ryszard Rozak <rrozak@antmicro.com>
antmicro · Nov 15, 2023 · 2f814bd · 2f814bd
1 parent bcd960d
commit 2f814bd
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/V3Randomize.cpp b/src/V3Randomize.cpp
@@ -370,7 +370,9 @@ class RandomizeVisitor final : public VNVisitor {
         auto* newp = VN_AS(m_memberMap.findMember(nodep, "new"), NodeFTask);
         UASSERT_OBJ(newp, nodep, "No new() in class");
         auto* taskp = newSetupConstraintsTask(nodep);
-        newp->addStmtsp(new AstTaskRef{fl, taskp->name(), nullptr});
+        AstTaskRef* const setupTaskRefp = new AstTaskRef{fl, taskp->name(), nullptr};
+        setupTaskRefp->taskp(taskp);
+        newp->addStmtsp(new AstStmtExpr{fl, setupTaskRefp});
 
         nodep->user1(false);
 
@@ -392,7 +394,7 @@ class RandomizeVisitor final : public VNVisitor {
                     auto* const methodp = new AstCMethodHard{
                         fl, new AstVarRef{fl, genp, VAccess::READWRITE}, "hard", exprp};
                     methodp->dtypeSetVoid();
-                    taskp->addStmtsp(methodp);
+                    taskp->addStmtsp(new AstStmtExpr{fl, methodp});
                 }
         });
     }