Commit fee55328 authored by Wenhao Chen's avatar Wenhao Chen Committed by binmakeswell
Browse files

[NFC] polish runtime_preparation_pass style (#4266)

parent 3883db45
......@@ -55,7 +55,7 @@ def size_processing(size: Union[int, torch.Size],
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
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