From a0bf9c043bf16e6ff18583134957921d5c44d979 Mon Sep 17 00:00:00 2001 From: Saoirse Stewart Date: Fri, 27 Jun 2025 11:57:44 +0100 Subject: [PATCH] Arm backend: Add intermediate tensor for ADD accumulator in op_index_tensor --- backends/arm/operators/op_index_tensor.py | 14 ++++++++++++-- 1 file changed, 12 insertions(+), 2 deletions(-) diff --git a/backends/arm/operators/op_index_tensor.py b/backends/arm/operators/op_index_tensor.py index 8c5c84ddd5a..36d0b37e090 100644 --- a/backends/arm/operators/op_index_tensor.py +++ b/backends/arm/operators/op_index_tensor.py @@ -189,11 +189,16 @@ def define_node( if i == 0: gather_index_name = reshaped_idxs.name else: + add_idxs = tosa_graph.addIntermediate( + reshaped_idxs.shape, + reshaped_idxs.dtype, + ) tosa_graph.addOperator( ts.TosaOp.Op().ADD, [gather_index_name, reshaped_idxs.name], - [gather_index_name], + [add_idxs.name], ) + gather_index_name = add_idxs.name gather_vals_shape = [N, K, C] reshaped_input = tosa_graph.addIntermediate(gather_vals_shape, values.dtype) @@ -314,11 +319,16 @@ def define_node( if i == 0: gather_index_name = reshaped_idxs.name else: + add_idxs = tosa_graph.addIntermediate( + reshaped_idxs.shape, + reshaped_idxs.dtype, + ) tosa_graph.addOperator( ts.TosaOp.Op().ADD, [gather_index_name, reshaped_idxs.name], - [gather_index_name], + [add_idxs.name], ) + gather_index_name = add_idxs.name gather_vals_shape = [N, K, C] reshaped_input = tosa_graph.addIntermediate(gather_vals_shape, values.dtype)