Skip to content

GitLab

Commit f566c9b9 authored by Zirui Zhu's avatar Zirui Zhu Committed by Frank Lee
Browse files

[NFC] polish colossalai/pipeline/utils.py code style (#1562)

parent e070ca45
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
colossalai/pipeline/utils.py
View file @ f566c9b9
...@@ -250,6 +250,7 @@ def call_module(module, args=None, kwargs=None): ...@@ -250,6 +250,7 @@ def call_module(module, args=None, kwargs=None):
else: else:
return module(*args_needed, **kwargs) return module(*args_needed, **kwargs)
def customized_partition(exec_seq): def customized_partition(exec_seq):
''' '''
This function will analyze the exec_seq. In the exec_seq, users will use 'SPLIT_NODE' as an This function will analyze the exec_seq. In the exec_seq, users will use 'SPLIT_NODE' as an
...@@ -269,4 +270,3 @@ def customized_partition(exec_seq): ...@@ -269,4 +270,3 @@ def customized_partition(exec_seq):
stop += 1 stop += 1
customized_parts[rank] = [(start, stop)] customized_parts[rank] = [(start, stop)]
return customized_parts return customized_parts
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment