Skip to content

Commit 7b82831

Browse files
committed
correcting rand test
1 parent 156263f commit 7b82831

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

tests/py/dynamo/conversion/test_rand_aten.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -74,10 +74,9 @@ def __init__(self, rand_op, size):
7474
def forward(self):
7575
return self.rand_op(self.size)
7676

77-
grid_model = TestModule(op, shape_or_input)
77+
rand_model = TestModule(op, shape_or_input)
7878
# cannot use self.run_test() since it expects input in form of tensor
7979

80-
# self.run_test(grid_model, None)
8180
fx_graph = torch.fx.symbolic_trace(grid_model)
8281
torch._dynamo.reset()
8382

0 commit comments

Comments
 (0)