From 8240c9deaf38b09d77efae6b135cad9ed630a09c Mon Sep 17 00:00:00 2001 From: CWHer Date: Tue, 18 Jul 2023 11:17:49 +0800 Subject: [PATCH] [NFC] polish runtime_preparation_pass style --- colossalai/auto_parallel/passes/runtime_preparation_pass.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/colossalai/auto_parallel/passes/runtime_preparation_pass.py b/colossalai/auto_parallel/passes/runtime_preparation_pass.py index 9a2314826448..1a6dc7815176 100644 --- a/colossalai/auto_parallel/passes/runtime_preparation_pass.py +++ b/colossalai/auto_parallel/passes/runtime_preparation_pass.py @@ -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.