Skip to content

GitLab

Unverified Commit c9e3ee38 authored by Zirui Zhu's avatar Zirui Zhu Committed by GitHub
Browse files

[NFC] polish colossalai/context/process_group_initializer/initializer_2d.py code style (#2726)

parent 1819373e
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 2 deletions
colossalai/context/process_group_initializer/initializer_2d.py
View file @ c9e3ee38
......@@ -2,10 +2,11 @@ import math
import torch.distributed as dist
from colossalai.global_variables import tensor_parallel_env as env
from colossalai.registry import DIST_GROUP_INITIALIZER
from .process_group_initializer import ProcessGroupInitializer
from ..parallel_mode import ParallelMode
from colossalai.global_variables import tensor_parallel_env as env
from .process_group_initializer import ProcessGroupInitializer
def _check_summa_env_var(summa_dim):
......
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