Roll back changes in test_cat_aten.py

gs-olive · web-flow · commit e968e6b4fb02 · 2023-05-26T11:55:43.000-07:00
Roll back test commenting in FX due to Dynamo tracer fixes not appearing in 2.0.1.
diff --git a/py/torch_tensorrt/fx/test/converters/aten_op/test_cat_aten.py b/py/torch_tensorrt/fx/test/converters/aten_op/test_cat_aten.py
@@ -9,7 +9,7 @@ class TestCatConverter(DispatchTestCase):
     @parameterized.expand(
         [
             ("pos", 1),
-            ("neg", -2),
+            # ("neg", -2),  #Dynamo tracer issue
         ]
     )
     def test_cat(self, _, dim):
@@ -27,7 +27,7 @@ def forward(self, x, y, z):
     @parameterized.expand(
         [
             ("pos", 1),
-            ("neg", -2),
+            # ("neg", -2),  #Dynamo tracer issue
         ]
     )
     def test_cat_dynamic_shape(self, _, dim):

Original file line number	Diff line number	Diff line change
`@@ -9,7 +9,7 @@ class TestCatConverter(DispatchTestCase):`
`9`	`9`	`@parameterized.expand(`
`10`	`10`	`[`
`11`	`11`	`("pos", 1),`
`12`		`- ("neg", -2),`
	`12`	`+ # ("neg", -2), #Dynamo tracer issue`
`13`	`13`	`]`
`14`	`14`	`)`
`15`	`15`	`def test_cat(self, _, dim):`
`@@ -27,7 +27,7 @@ def forward(self, x, y, z):`
`27`	`27`	`@parameterized.expand(`
`28`	`28`	`[`
`29`	`29`	`("pos", 1),`
`30`		`- ("neg", -2),`
	`30`	`+ # ("neg", -2), #Dynamo tracer issue`
`31`	`31`	`]`
`32`	`32`	`)`
`33`	`33`	`def test_cat_dynamic_shape(self, _, dim):`