github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-08-11 12:51:55 +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
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 6 additions and 5 deletions
Show Stats 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']