Include variable to map if it uses rand mode

Signed-off-by: Ryszard Rozak <[email protected]>
antmicro · Dec 6, 2024 · 89fb5a2 · 89fb5a2
1 parent ee69b65
commit 89fb5a2
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/src/V3Randomize.cpp b/src/V3Randomize.cpp
@@ -1163,10 +1163,8 @@ class RandomizeVisitor final : public VNVisitor {
     }
     void fillConstrainedMapRecurse(AstNode* const nodep, AstClass* const classp) {
         if (AstVarRef* const varRefp = VN_CAST(nodep, VarRef)) {
-            if (varRefp->user1()) {
-                AstVar* const varp = varRefp->varp();
-                m_constrainedVars[classp].insert(varp);
-            }
+            AstVar* const varp = varRefp->varp();
+            if (varRefp->user1() || varp->user1()) m_constrainedVars[classp].insert(varp);
             return;
         }
         if (nodep->op1p()) fillConstrainedMapRecurse(nodep->op1p(), classp);