From 49ba619085c33eef372e73b6a45aecdc3d37937f Mon Sep 17 00:00:00 2001 From: oahzxl Date: Fri, 23 Dec 2022 14:26:43 +0800 Subject: [PATCH] code style --- chunk_codegen.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/chunk_codegen.py b/chunk_codegen.py index e3a7643d7..40196285e 100644 --- a/chunk_codegen.py +++ b/chunk_codegen.py @@ -1899,7 +1899,7 @@ def emit_code_with_chunk( chunk_outputs = [i["outputs"][0] for i in chunk_search] chunk_outputs_dim = [i["outputs_dim"] for i in chunk_search] - + chunk_prepose_nodes = [i["args"]["prepose_nodes"] for i in chunk_search] node_idx = 0 @@ -1933,7 +1933,9 @@ def emit_code_with_chunk( emit_node_func(node, body) # replace input var with chunk var for input_node_idx, input_node in enumerate(chunk_inputs[region_idx]): - for idx, dim in chunk_inputs_dim[region_idx][input_node_idx].items(): + for idx, dim in chunk_inputs_dim[region_idx][ + input_node_idx + ].items(): if idx == node_idx: chunk_slice = _gen_chunk_slice_dim( dim, "chunk_idx", _get_node_shape(input_node)