Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
OpenDAS
ColossalAI
Commits
786a398a
Commit
786a398a
authored
Dec 23, 2022
by
oahzxl
Browse files
code style
parent
9b1b8903
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
4 additions
and
2 deletions
+4
-2
chunk_codegen.py
chunk_codegen.py
+4
-2
No files found.
chunk_codegen.py
View file @
786a398a
...
@@ -920,9 +920,13 @@ class IndexTracer(object):
...
@@ -920,9 +920,13 @@ class IndexTracer(object):
# loop cur node's all arg until out of chunk
# loop cur node's all arg until out of chunk
while
len
(
tmp_cur_prepose_nodes
)
>
0
:
while
len
(
tmp_cur_prepose_nodes
)
>
0
:
if
prepose_flag
==
False
:
break
tmp_next_prepose_nodes
=
[]
tmp_next_prepose_nodes
=
[]
tmp_cur_related_prepose_nodes
.
extend
(
tmp_cur_prepose_nodes
)
tmp_cur_related_prepose_nodes
.
extend
(
tmp_cur_prepose_nodes
)
for
cur_prepose_node
in
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
:
for
cur_prepose_node_arg
in
cur_prepose_node
.
args
:
if
type
(
cur_prepose_node_arg
)
!=
type
(
cur_prepose_node
):
if
type
(
cur_prepose_node_arg
)
!=
type
(
cur_prepose_node
):
continue
continue
...
@@ -942,8 +946,6 @@ class IndexTracer(object):
...
@@ -942,8 +946,6 @@ class IndexTracer(object):
else
:
else
:
prepose_flag
=
False
prepose_flag
=
False
break
break
break
break
# non compute op
# non compute op
else
:
else
:
tmp_next_prepose_nodes
.
append
(
cur_prepose_node_arg
)
tmp_next_prepose_nodes
.
append
(
cur_prepose_node_arg
)
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment