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
0b6af554
Commit
0b6af554
authored
Jan 09, 2023
by
oahzxl
Browse files
remove useless function
parent
d914a21d
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
0 additions
and
13 deletions
+0
-13
colossalai/autochunk/trace_indice.py
colossalai/autochunk/trace_indice.py
+0
-13
No files found.
colossalai/autochunk/trace_indice.py
View file @
0b6af554
...
@@ -10,7 +10,6 @@ class TraceIndice(object):
...
@@ -10,7 +10,6 @@ class TraceIndice(object):
def
__init__
(
self
,
node_list
)
->
None
:
def
__init__
(
self
,
node_list
)
->
None
:
self
.
node_list
=
node_list
self
.
node_list
=
node_list
self
.
indice_trace_list
=
self
.
_init_indice_trace_list
()
self
.
indice_trace_list
=
self
.
_init_indice_trace_list
()
self
.
indice_trace_equal
=
[]
self
.
indice_view_list
=
{}
self
.
indice_view_list
=
{}
self
.
indice_count
=
-
1
self
.
indice_count
=
-
1
...
@@ -486,18 +485,6 @@ class TraceIndice(object):
...
@@ -486,18 +485,6 @@ class TraceIndice(object):
}
}
self
.
indice_view_list
[
node
]
=
view_dict
self
.
indice_view_list
[
node
]
=
view_dict
def
_merge_equal_idx
(
self
):
idx_equal
=
copy
.
deepcopy
(
self
.
indice_trace_equal
)
idx_equal
.
reverse
()
for
idx
in
idx_equal
:
merge_to
=
min
(
idx
)
merge_from
=
max
(
idx
)
for
trace
in
self
.
indice_trace_list
:
if
merge_from
in
trace
[
"indice"
]:
trace
[
"indice"
]
=
[
merge_to
if
i
==
merge_from
else
i
for
i
in
trace
[
"indice"
]
]
def
trace_indice
(
self
):
def
trace_indice
(
self
):
for
idx
,
node
in
enumerate
(
self
.
node_list
):
for
idx
,
node
in
enumerate
(
self
.
node_list
):
if
node
.
op
==
"placeholder"
:
if
node
.
op
==
"placeholder"
:
...
...
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