Skip to content

GitLab

"lib/llm/src/vscode:/vscode.git/clone" did not exist on "b2aa2317041ed79d390874a8cf89d48452779ddf"
Commit 1168b50e authored by Zirui Zhu's avatar Zirui Zhu Committed by binmakeswell
Browse files

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

parent 196d4696
Show whitespace changes
Inline Side-by-side
Showing with 7 additions and 6 deletions
colossalai/engine/schedule/_pipeline_schedule_v2.py
View file @ 1168b50e
#!/usr/bin/env python #!/usr/bin/env python
# -*- encoding: utf-8 -*- # -*- 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 torch.cuda
import colossalai.communication.p2p_v2 as comm
from colossalai import engine
from colossalai.context.parallel_mode import ParallelMode from colossalai.context.parallel_mode import ParallelMode
from colossalai.core import global_context as gpc from colossalai.core import global_context as gpc
from colossalai.utils.cuda import get_current_device from colossalai.utils.cuda import get_current_device
......
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