diff --git a/test/dynamo/test_dynamo.py b/test/dynamo/test_dynamo.py index 58c4b83b672..9584f8ab57c 100644 --- a/test/dynamo/test_dynamo.py +++ b/test/dynamo/test_dynamo.py @@ -588,7 +588,7 @@ def foo(x): def test_new_with_sizes(self): def foo(x): - return x.new(*x.sizes()) + x + return x.new(*x.size()) + x optfoo = torch.compile(backend="openxla")(foo)