YH 2629f9717d [tensor] Refactor handle_trans_spec in DistSpecManager		2023-05-06 17:55:37 +08:00
..
d_tensor	[doc] Fix typo under colossalai and doc(#3618)	2023-04-26 11:38:43 +08:00
__init__.py	[Gemini] ParamOpHook -> ColoParamOpHook (#2080)	2022-12-05 17:11:06 +08:00
colo_parameter.py	[polish] polish ColoTensor and its submodules (#2537)	2023-02-03 11:44:10 +08:00
colo_tensor.py	[doc] Fix typo under colossalai and doc(#3618)	2023-04-26 11:38:43 +08:00
comm_spec.py	[doc] Fix typo under colossalai and doc(#3618)	2023-04-26 11:38:43 +08:00
compute_spec.py	[doc] Fix typo under colossalai and doc(#3618)	2023-04-26 11:38:43 +08:00
const.py	…
dist_spec_mgr.py	[tensor] Refactor handle_trans_spec in DistSpecManager	2023-05-06 17:55:37 +08:00
distspec.py	[doc] Fix typo under colossalai and doc(#3618)	2023-04-26 11:38:43 +08:00
op_wrapper.py	[doc] update rst and docstring (#1351)	2022-07-21 15:54:53 +08:00
param_op_hook.py	Fix typo (#3448)	2023-04-06 09:43:31 +08:00
process_group.py	[polish] polish ColoTensor and its submodules (#2537)	2023-02-03 11:44:10 +08:00
shape_consistency.py	[doc] Fix typo under colossalai and doc(#3618)	2023-04-26 11:38:43 +08:00
sharding_spec.py	[doc] Fix typo under colossalai and doc(#3618)	2023-04-26 11:38:43 +08:00
tensor_spec.py	[autoparallel] fix bugs caused by negative dim key (#1808)	2022-11-08 17:03:50 +08:00
utils.py	[doc] Fix typo under colossalai and doc(#3618)	2023-04-26 11:38:43 +08:00