correcting rand test

pytorch · Apr 4, 2024 · 7ce66ea · 7ce66ea
1 parent 123f8b6
commit 7ce66ea
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/tests/py/dynamo/conversion/test_rand_aten.py b/tests/py/dynamo/conversion/test_rand_aten.py
@@ -74,10 +74,9 @@ def __init__(self, rand_op, size):
             def forward(self):
                 return self.rand_op(self.size)
 
-        grid_model = TestModule(op, shape_or_input)
+        rand_model = TestModule(op, shape_or_input)
         # cannot use self.run_test() since it expects input in form of tensor
 
-        # self.run_test(grid_model, None)
         fx_graph = torch.fx.symbolic_trace(grid_model)
         torch._dynamo.reset()