github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

[hotfix] got sliced types (#1614)

pull/1615/head
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 self.device_mesh = device_mesh
@abstractmethod @abstractmethod
def generate(self, operand_mapping: Dict[str:Operand]) -> List[ShardingStrategy_V2]: def generate(self, operand_mapping: Dict[str, Operand]) -> List[ShardingStrategy_V2]:
""" """
""" """
pass pass