Skip to content

Commit 47842b0

Browse files
Arm backend: Fix broken SD3Transformer2DModel test (#16247)
Fixes broken SD3Transformer2DModel vgf tests. cc @freddan80 @per @zingo @digantdesai Signed-off-by: Oscar Andersson <oscar.andersson@arm.com>
1 parent d412189 commit 47842b0

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

backends/arm/test/models/stable_diffusion/test_SD3Transformer2DModel.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -43,7 +43,10 @@ class TestSD3Transformer2DModel:
4343
"executorch_exir_dialects_edge__ops_aten_permute_copy_default": 1,
4444
}
4545

46-
ops_after_partitioner_vgf_quantize = ops_after_partitioner_FP
46+
ops_after_partitioner_vgf_quantize = {
47+
"executorch_exir_dialects_edge__ops_dim_order_ops__to_dim_order_copy_default": 1,
48+
"torch.ops.higher_order.executorch_call_delegate": 1,
49+
}
4750
ops_after_partitioner_vgf_no_quantize = ops_after_partitioner_FP
4851

4952
def _prepare_inputs(

0 commit comments

Comments
 (0)