From 786a398a6bdea395e2ca8ddde87c87c8470d971b Mon Sep 17 00:00:00 2001 From: oahzxl Date: Fri, 23 Dec 2022 17:42:51 +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 9623a9d9b..f87a3a132 100644 --- a/chunk_codegen.py +++ b/chunk_codegen.py @@ -920,9 +920,13 @@ class IndexTracer(object): # loop cur node's all arg until out of chunk while len(tmp_cur_prepose_nodes) > 0: + if prepose_flag == False: + break tmp_next_prepose_nodes = [] tmp_cur_related_prepose_nodes.extend(tmp_cur_prepose_nodes) for cur_prepose_node in tmp_cur_prepose_nodes: + if prepose_flag == False: + break for cur_prepose_node_arg in cur_prepose_node.args: if type(cur_prepose_node_arg) != type(cur_prepose_node): continue @@ -942,8 +946,6 @@ class IndexTracer(object): else: prepose_flag = False break - break - break # non compute op else: tmp_next_prepose_nodes.append(cur_prepose_node_arg)