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

[NFC] posh colossalai/context/process_group_initializer/initializer_sequence.py code style (#2712) 

Co-authored-by: Ziyue Jiang <ziyue.jiang@gmail.com>
pull/2720/head
Ziyue Jiang 2 years ago committed by GitHub
parent
b3d10db5f1
commit
4603538ddd
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 1 deletions
Whitespace
Show all changes Ignore whitespace when comparing lines Ignore changes in amount of whitespace Ignore changes in whitespace at EOL
Unified View
Diff Options
Show Stats Download Patch File Download Diff File
  1. 3
      colossalai/context/process_group_initializer/initializer_sequence.py

3
colossalai/context/process_group_initializer/initializer_sequence.py
Unescape View File

@ -3,9 +3,10 @@
import torch.distributed as dist import torch.distributed as dist
from colossalai.registry import DIST_GROUP_INITIALIZER from colossalai.registry import DIST_GROUP_INITIALIZER
from ..parallel_mode import ParallelMode
from .initializer_tensor import Initializer_Tensor from .initializer_tensor import Initializer_Tensor
from .process_group_initializer import ProcessGroupInitializer from .process_group_initializer import ProcessGroupInitializer
from ..parallel_mode import ParallelMode
@DIST_GROUP_INITIALIZER.register_module @DIST_GROUP_INITIALIZER.register_module

Write Preview
Loading…
Cancel
Save