Skip to content

Commit b903c30

Browse files
authored
Arm backend: Fix vgf_quant swin test op-count and test vgf models in trunk job. (pytorch#19841)
1 parent 96b19af commit b903c30

2 files changed

Lines changed: 5 additions & 1 deletion

File tree

.github/workflows/trunk.yml

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -278,6 +278,7 @@ jobs:
278278
matrix:
279279
include:
280280
- test_arm_backend: test_pytest_ops_vkml
281+
- test_arm_backend: test_pytest_models_vkml
281282
- test_arm_backend: test_ootb_tests_vgf
282283
fail-fast: false
283284
with:

backends/arm/test/models/test_swin2sr_arm.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -42,6 +42,9 @@
4242
"executorch_exir_dialects_edge__ops_quantized_decomposed_dequantize_per_tensor_default": 5,
4343
"executorch_exir_dialects_edge__ops_quantized_decomposed_quantize_per_tensor_default": 6,
4444
}
45+
swin2sr_vgf_quant_lowered_outer_graph_ops = {
46+
"torch.ops.higher_order.executorch_call_delegate": 1,
47+
}
4548

4649

4750
class TinySwin2SR(torch.nn.Module):
@@ -110,7 +113,7 @@ def test_swin2sr_vgf_quant():
110113
quantize=True,
111114
run_on_vulkan_runtime=sys.platform == "linux",
112115
)
113-
pipeline.change_args("check_count.exir", swin2sr_int_lowered_outer_graph_ops)
116+
pipeline.change_args("check_count.exir", swin2sr_vgf_quant_lowered_outer_graph_ops)
114117
pipeline.run()
115118

116119

0 commit comments

Comments
 (0)