Update test_reshape_aten.py

frank-wei · web-flow · commit a76099403e97 · 2022-12-16T22:48:20.000-08:00
diff --git a/py/torch_tensorrt/fx/test/converters/aten_op/test_reshape_aten.py b/py/torch_tensorrt/fx/test/converters/aten_op/test_reshape_aten.py
@@ -25,7 +25,7 @@ def forward(self, x):
         self.run_test(
             TestModule(target_shape),
             inputs,
-            expected_ops={torch.ops.aten._reshape_alias.default},
+            expected_ops={torch.ops.aten.view.default},
         )
 
     ## TODO: proxytensor tracer does not support output size containing -1. If dim=0 is set to -1 for dynamic batch,

Original file line number	Diff line number	Diff line change
`@@ -25,7 +25,7 @@ def forward(self, x):`
`25`	`25`	`self.run_test(`
`26`	`26`	`TestModule(target_shape),`
`27`	`27`	`inputs,`
`28`		`- expected_ops={torch.ops.aten._reshape_alias.default},`
	`28`	`+ expected_ops={torch.ops.aten.view.default},`
`29`	`29`	`)`
`30`	`30`
`31`	`31`	`## TODO: proxytensor tracer does not support output size containing -1. If dim=0 is set to -1 for dynamic batch,`