YuliangLiu0306 33f0744d51 [tensor] add shape consistency feature to support auto spec transform (#1418) ... * [tensor] add shape consistency feature to supportauto sharding spec transform. * [tensor] remove unused argument in simulator, add doc string for target pair.		2022-08-10 11:29:17 +08:00
..
__init__.py	…
colo_parameter.py	…
colo_tensor.py	…
compute_spec.py	…
const.py	…
dist_spec_mgr.py	…
distspec.py	…
op_wrapper.py	…
param_op_hook.py	…
process_group.py	[hotfix] adapt ProcessGroup and Optimizer to ColoTensor (#1388)	2022-07-29 19:33:24 +08:00
shape_consistency.py	[tensor] add shape consistency feature to support auto spec transform (#1418)	2022-08-10 11:29:17 +08:00
sharding_spec.py	[tensor] add shape consistency feature to support auto spec transform (#1418)	2022-08-10 11:29:17 +08:00
tensor_spec.py	…
utils.py	…