Skip to content

Commit ae755b5

Browse files
committed
Arm backend: Dont try to fuse const for TOSA ops
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. Signed-off-by: Per Åstrand <[email protected]> Change-Id: I18af1ef2ceb2d262f66fff7df6b019782b79b767
1 parent 3dac421 commit ae755b5

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)