[quantizer] fix logic for new_indices (#377)

alibaba · Dec 2, 2024 · 2d7f919 · 2d7f919
1 parent 19e5f6d
commit 2d7f919
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tinynn/graph/quantization/quantizer.py b/tinynn/graph/quantization/quantizer.py
@@ -2949,9 +2949,9 @@ def _is_broadcastable_binary_quantized_op_node(node: TraceNode, custom_data) ->
                     if node.prev_indices[i] is None:
                         new_indices.append(i)
                     elif isinstance(node.prev_indices[i], (tuple, list)):
-                        new_indices.push(node.prev_indices[i] + [i])
+                        new_indices.append(node.prev_indices[i] + [i])
                     else:
-                        new_indices.push([node.prev_indices[i], i])
+                        new_indices.append([node.prev_indices[i], i])
                 with override_current_trace_graph(graph):
                     trace_func = TraceFunction('torch.broadcast_tensors', False, prefix='fuse_').parse_args(
                         node.prev_tensors[0], node.prev_tensors[1]