github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-06-26 15:32:22 +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
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats 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