Skip to content

Commit 37c1958

Browse files
authored
Arm backend: Dont try to fuse const for TOSA ops (#13575)
### Summary The TOSA ops doesn't have any backed reference implementation, so avoid trying to fuse constant ops and leave for the backend compiler to do the work. ### Test plan Tested through unit tests. cc @digantdesai @freddan80 @zingo @oscarandersson8218 Signed-off-by: Per Åstrand <[email protected]>
1 parent def4c64 commit 37c1958

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

backends/arm/_passes/fuse_constant_ops_pass.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -107,7 +107,11 @@ def call(self, graph_module):
107107
for node in graph_module.graph.nodes:
108108
if node.op != "call_function":
109109
continue
110-
if node.target == exir_ops.backend.tosa.TABLE.default:
110+
if node.target in [
111+
exir_ops.backend.tosa.TABLE.default,
112+
exir_ops.backend.tosa.RESCALE.default,
113+
exir_ops.backend.tosa.TRANSPOSE.default,
114+
]:
111115
continue
112116

113117
input_nodes = node.all_input_nodes

0 commit comments

Comments
 (0)