Fix cat tests (pytorch#1086)

Summary: Pull Request resolved: pytorch#1086 Reviewed By: mcr229 Differential Revision: D50504935 fbshipit-source-id: 6b541d0558f1cb1ea99893a3cfa96ba35161c5d8
kirklandsign · Oct 24, 2023 · 427641c · 427641c
1 parent c6c2ea9
commit 427641c
Showing 1 changed file with 12 additions and 11 deletions.
diff --git a/backends/xnnpack/test/ops/cat.py b/backends/xnnpack/test/ops/cat.py
@@ -21,17 +21,18 @@ def forward(self, x, y):
     def test_fp32_cat(self):
         inputs = (torch.ones(1, 2, 3), torch.ones(3, 2, 3))
         (
-            Tester(self.Cat(), inputs).export()
-            # .check_count({"torch.ops.aten.cat": 1})
-            # .to_edge()
-            # .check_count({"executorch_exir_dialects_edge__ops_aten_cat": 1})
-            # .partition()
-            # .check_count({"torch.ops.executorch_call_delegate": 1})
-            # .check_not(["executorch_exir_dialects_edge__ops_aten_cat"])
-            # .to_executorch()
-            # .serialize()
-            # .run_method()
-            # .compare_outputs()
+            Tester(self.Cat(), inputs)
+            .export()
+            .check_count({"torch.ops.aten.cat": 1})
+            .to_edge()
+            .check_count({"executorch_exir_dialects_edge__ops_aten_cat": 1})
+            .partition()
+            .check_count({"torch.ops.executorch_call_delegate": 1})
+            .check_not(["executorch_exir_dialects_edge__ops_aten_cat"])
+            .to_executorch()
+            .serialize()
+            .run_method()
+            .compare_outputs()
         )
 
     class CatNegativeDim(torch.nn.Module):