Minor updates to xnnpack tests

tugsbayasgalan · facebook-github-bot · commit 44ec67d17251 · 2024-04-29T19:25:09.000-07:00
Summary: This makes torch.export migration easier

Reviewed By: mcr229

Differential Revision: D56720545

fbshipit-source-id: fa6fc1bd770432201e7c39fafdb45f8a28392dbd
diff --git a/backends/xnnpack/test/ops/add.py b/backends/xnnpack/test/ops/add.py
@@ -95,8 +95,7 @@ def test_qs8_add_constant(self):
             .check_not(["executorch_exir_dialects_edge__ops_aten_add_Tensor"])
             .to_executorch()
             .serialize()
-            .run_method()
-            .compare_outputs()
+            .run_method_compare_outputs()
         )
 
     def test_qs8_add(self):
diff --git a/backends/xnnpack/test/tester/tester.py b/backends/xnnpack/test/tester/tester.py
@@ -189,7 +189,7 @@ def run(
     ) -> None:
         self.exported_program = export(
             artifact, inputs, dynamic_shapes=self.dynamic_shapes
-        )
+        ).run_decompositions()
 
     @property
     def artifact(self) -> ExportedProgram:

Original file line number	Diff line number	Diff line change
`@@ -95,8 +95,7 @@ def test_qs8_add_constant(self):`
`95`	`95`	`.check_not(["executorch_exir_dialects_edge__ops_aten_add_Tensor"])`
`96`	`96`	`.to_executorch()`
`97`	`97`	`.serialize()`
`98`		`- .run_method()`
`99`		`- .compare_outputs()`
	`98`	`+ .run_method_compare_outputs()`
`100`	`99`	`)`
`101`	`100`
`102`	`101`	`def test_qs8_add(self):`