@@ -80,17 +80,17 @@ def name_elems(self):
8080 for i ,f in enumerate (self .field_l ):
8181 f .name = self .name + "[" + str (i ) + "]"
8282
83- def pre_randomize (self ):
83+ def pre_randomize (self , visited ):
8484 # Set the size field for arrays that don't
8585 # have a random size
8686 if self .is_rand_sz :
8787 self .size .set_used_rand (True )
8888 else :
8989 self ._set_size (len (self .field_l ))
90- FieldCompositeModel .pre_randomize (self )
90+ FieldCompositeModel .pre_randomize (self , visited )
9191
92- def post_randomize (self ):
93- FieldCompositeModel .post_randomize (self )
92+ def post_randomize (self , visited ):
93+ FieldCompositeModel .post_randomize (self , visited )
9494 self .sum_expr = None
9595 self .sum_expr_btor = None
9696
@@ -116,13 +116,9 @@ def build(self, builder):
116116 self ._set_size (len (self .field_l ))
117117 super ().build (builder )
118118
119- # def set_used_rand(self, is_rand, level=0):
120- # if self.is_rand_sz:
121- # self.size.set_used_rand(is_rand)
122- # FieldCompositeModel.set_used_rand(self, is_rand, level=level)
123- def set_used_rand (self , is_rand , level = 0 ):
124- super ().set_used_rand (is_rand , level )
125- self .size .set_used_rand (is_rand , level + 1 )
119+ def set_used_rand (self , is_rand , level = 0 , in_set = None ):
120+ super ().set_used_rand (is_rand , level , in_set )
121+ self .size .set_used_rand (is_rand , level + 1 , in_set )
126122
127123 def get_sum_expr (self ):
128124 if self .sum_expr is None :
0 commit comments