github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-09-08 04:24:47 +00:00
Code Issues Packages Projects Releases Wiki Activity

[hotfix] got sliced types (#1614)

Browse Source
This commit is contained in:
YuliangLiu0306
2022-09-20 14:32:42 +08:00
committed by GitHub
parent d397842fa8
commit 6a8f8cc05e
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
colossalai/auto_parallel/solver/sharding_strategy.py
View File

@@ -105,7 +105,7 @@ class StrategyGenerator_V2(ABC):
self.device_mesh = device_mesh
@abstractmethod
def generate(self, operand_mapping: Dict[str:Operand]) -> List[ShardingStrategy_V2]:
def generate(self, operand_mapping: Dict[str, Operand]) -> List[ShardingStrategy_V2]:
"""
"""
pass