Skip to content

GitLab

"tests/git@developer.sourcefind.cn:OpenDAS/colossalai.git" did not exist on "0fcfb1e00d71fe3eede020125dc4d9b3970d7e07"
Commit fee55328 authored by Wenhao Chen's avatar Wenhao Chen Committed by binmakeswell
Browse files

[NFC] polish runtime_preparation_pass style (#4266)

parent 3883db45
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
colossalai/auto_parallel/passes/runtime_preparation_pass.py
View file @ fee55328
...@@ -55,7 +55,7 @@ def size_processing(size: Union[int, torch.Size], ...@@ -55,7 +55,7 @@ def size_processing(size: Union[int, torch.Size],
def solution_annotation_pass(gm: torch.fx.GraphModule, solution: List[int], def solution_annotation_pass(gm: torch.fx.GraphModule, solution: List[int],
strategies_constructor: StrategiesConstructor): strategies_constructor: StrategiesConstructor):
""" """
This method is used to stick the solution strategy to the nodes and add the information This method is used to stick the solution strategy to the nodes and add the information
required in runtime into graph as placeholder nodes. required in runtime into graph as placeholder nodes.
......
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