From fee553288b7b4f3116a8cc4d1a5052183302c709 Mon Sep 17 00:00:00 2001 From: Wenhao Chen Date: Tue, 18 Jul 2023 11:54:09 +0800 Subject: [PATCH] [NFC] polish runtime_preparation_pass style (#4266) --- 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 9a2314826..1a6dc7815 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.