Browse Source

[hotfix] got sliced types (#1614)

pull/1615/head
YuliangLiu0306 2 years ago committed by GitHub
parent
commit
6a8f8cc05e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
  1. 2
      colossalai/auto_parallel/solver/sharding_strategy.py

2
colossalai/auto_parallel/solver/sharding_strategy.py

@ -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

Loading…
Cancel
Save