Browse Source

[NFC] polish pipeline process group code style (#2694)

pull/2708/head
アマデウス 2 years ago committed by GitHub
parent
commit
534f68c83c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
  1. 3
      colossalai/context/process_group_initializer/initializer_pipeline.py

3
colossalai/context/process_group_initializer/initializer_pipeline.py

@ -4,8 +4,9 @@
from torch import distributed as dist from torch import distributed as dist
from colossalai.registry import DIST_GROUP_INITIALIZER from colossalai.registry import DIST_GROUP_INITIALIZER
from .process_group_initializer import ProcessGroupInitializer
from ..parallel_mode import ParallelMode from ..parallel_mode import ParallelMode
from .process_group_initializer import ProcessGroupInitializer
@DIST_GROUP_INITIALIZER.register_module @DIST_GROUP_INITIALIZER.register_module

Loading…
Cancel
Save