Fill setup task body

Signed-off-by: Ryszard Rozak <[email protected]>
antmicro · Dec 3, 2024 · 7e204ee · 7e204ee
1 parent 0b53f4e
commit 7e204ee
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/src/V3Randomize.cpp b/src/V3Randomize.cpp
@@ -886,6 +886,11 @@ class ConstraintExprVisitor final : public VNVisitor {
         m_taskp = new AstTask{nodep->fileline(), name + "_setup_constraint", m_genp};
         nodep->user2p(m_taskp);
         iterateChildren(nodep);
+        if (AstConstraint* constrp = VN_CAST(nodep, Constraint)) {
+            m_taskp->addStmtsp(constrp->itemsp()->unlinkFrBackWithNext());
+        } else if (AstWith* withp = VN_CAST(nodep, With)) {
+            m_taskp->addStmtsp(withp->exprp()->unlinkFrBackWithNext());
+        }
     }
 };