Update test_flatten_aten.py

frank-wei · web-flow · commit 15653073b464 · 2022-12-16T22:47:39.000-08:00
diff --git a/py/torch_tensorrt/fx/test/converters/aten_op/test_flatten_aten.py b/py/torch_tensorrt/fx/test/converters/aten_op/test_flatten_aten.py
@@ -27,7 +27,7 @@ def forward(self, x):
         self.run_test(
             Flatten(start_dim, end_dim),
             inputs,
-            expected_ops={torch.ops.aten._reshape_alias.default},
+            expected_ops={torch.ops.aten.view.default},
             test_implicit_batch_dim=(start_dim != 0),
         )
 

Original file line number	Diff line number	Diff line change
`@@ -27,7 +27,7 @@ def forward(self, x):`
`27`	`27`	`self.run_test(`
`28`	`28`	`Flatten(start_dim, end_dim),`
`29`	`29`	`inputs,`
`30`		`- expected_ops={torch.ops.aten._reshape_alias.default},`
	`30`	`+ expected_ops={torch.ops.aten.view.default},`
`31`	`31`	`test_implicit_batch_dim=(start_dim != 0),`
`32`	`32`	`)`
`33`	`33`