github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-09-26 20:23:26 +00:00
Code Issues Packages Projects Releases Wiki Activity

[NFC] polish colossalai/auto_parallel/tensor_shard/deprecated/op_handler/strategy_generator.py code style (#2695)

Browse Source 
Co-authored-by: shenggan <csg19971016@gmail.com>
This commit is contained in:
Liu Ziming
2023-02-14 21:30:25 +08:00
committed by GitHub
parent 534f68c83c
commit 6427c406cf
1 changed files with 6 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
colossalai/auto_parallel/tensor_shard/deprecated/op_handler/strategy_generator.py
View File

@@ -1,6 +1,7 @@
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']