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

[NFC] polish colossalai/engine/schedule/_pipeline_schedule_v2.py code style (#3275)

pull/3313/head
Zirui Zhu 2023-03-28 14:31:38 +08:00 committed by binmakeswell
parent 196d4696d0
commit 1168b50e33
1 changed files with 7 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
colossalai/engine/schedule/_pipeline_schedule_v2.py
View File

@ -1,11 +1,12 @@
#!/usr/bin/env python
# -*- encoding: utf-8 -*-
from typing import Tuple, Iterable
from typing import Iterable, Tuple
from colossalai import engine
import colossalai.communication.p2p_v2 as comm
import torch.cuda
import colossalai.communication.p2p_v2 as comm
from colossalai import engine
from colossalai.context.parallel_mode import ParallelMode
from colossalai.core import global_context as gpc
from colossalai.utils.cuda import get_current_device