Skip to content

GitLab

Unverified Commit 6427c406 authored by Liu Ziming's avatar Liu Ziming Committed by GitHub
Browse files

[NFC] polish... 


[NFC] polish colossalai/auto_parallel/tensor_shard/deprecated/op_handler/strategy_generator.py code style (#2695)
Co-authored-by: default avatarshenggan <csg19971016@gmail.com>
parent 534f68c8
Show whitespace changes
Inline Side-by-side
Showing with 6 additions and 5 deletions
colossalai/auto_parallel/tensor_shard/deprecated/op_handler/strategy_generator.py
View file @ 6427c406
from dataclasses import dataclass
from abc import ABC, abstractmethod from abc import ABC, abstractmethod
from typing import List, Dict from dataclasses import dataclass
from typing import Dict, List
from colossalai.device.device_mesh import DeviceMesh from colossalai.device.device_mesh import DeviceMesh
__all__ = ['IntermediateStrategy', 'StrategyGenerator'] __all__ = ['IntermediateStrategy', 'StrategyGenerator']
......
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