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

Loading…
Cancel
Save