Reduce number of delegates in test_llama_arm.py

mansnils · mansnils · commit e7c117bcde78 · 2025-02-25T19:00:21.000+01:00
diff --git a/backends/arm/test/models/test_llama_arm.py b/backends/arm/test/models/test_llama_arm.py
@@ -117,7 +117,7 @@ def test_llama_tosa_MI(self):
                 .to_edge_transform_and_lower(
                     edge_compile_config=self._edge_compile_config
                 )
-                .check_count({"torch.ops.higher_order.executorch_call_delegate": 25})
+                .check_count({"torch.ops.higher_order.executorch_call_delegate": 14})
                 .to_executorch()
                 .run_method_and_compare_outputs(
                     inputs=llama_inputs, atol=1.8, rtol=0.01

Original file line number	Diff line number	Diff line change
`@@ -117,7 +117,7 @@ def test_llama_tosa_MI(self):`
`117`	`117`	`.to_edge_transform_and_lower(`
`118`	`118`	`edge_compile_config=self._edge_compile_config`
`119`	`119`	`)`
`120`		`- .check_count({"torch.ops.higher_order.executorch_call_delegate": 25})`
	`120`	`+ .check_count({"torch.ops.higher_order.executorch_call_delegate": 14})`
`121`	`121`	`.to_executorch()`
`122`	`122`	`.run_method_and_compare_outputs(`
`123`	`123`	`inputs=llama_inputs, atol=1.8, rtol=0.01`