github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

[NFC] polish runtime_preparation_pass style (#4266)

pull/4338/head
Wenhao Chen 2023-07-18 11:54:09 +08:00 committed by binmakeswell
parent 3883db452c
commit fee553288b
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
colossalai/auto_parallel/passes/runtime_preparation_pass.py
View File

@ -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.