ColossalAI/colossalai/tensor/d_tensor/misc.py
YuliangLiu0306 cd2b0eaa8d
[DTensor] refactor sharding spec (#2987)
* [autoparallel] refactor sharding spec

* rename function name
2023-03-07 11:08:11 +08:00

15 lines
231 B
Python

class LayoutException(Exception):
pass
class DuplicatedShardingDimensionError(LayoutException):
pass
class ShardingNotDivisibleError(LayoutException):
pass
class ShardingOutOfIndexError(LayoutException):
pass