diff --git a/test/dynamo/test_dynamo.py b/test/dynamo/test_dynamo.py index 34a37c710cc..d6003259c30 100644 --- a/test/dynamo/test_dynamo.py +++ b/test/dynamo/test_dynamo.py @@ -637,7 +637,7 @@ class DynamoOperationsTests(test_utils.XlaTestCase): 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)