Skip to content

Commit c3d303f

Browse files
committed
[arm][ez] Do not decompose linalg_vector_norm
Summary: ## Context #9938 made it so that `linalg_vector_norm` is now decomposed when exporting to Edge. However, this broke some tests in the arm delegate which does not expect the op to be decomposed. ## Changes Add `linalg_vector_norm` to the list of ops not to decompose in the arm partitioner. Test Plan: ## Test Plan Check CI that failing test is recovered.
1 parent 0bc52fe commit c3d303f

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

backends/arm/tosa_partitioner.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -173,6 +173,7 @@ def filter_fn(node: torch.fx.Node) -> bool:
173173
torch.ops.aten.upsample_nearest2d.vec,
174174
torch.ops.aten.eye.default,
175175
torch.ops.aten.linspace.default,
176+
torch.ops.aten.linalg_vector_norm.default,
176177
] + ops_to_not_decompose_if_quant_op
177178

178179
return (ops_to_not_decompose, filter_fn)

0 commit comments

Comments
 (0)