ColossalAI/colossalai/tensor
Frank Lee 190a6ea9c2
[dtensor] fixed readme file name and removed deprecated file (#4162)
2023-07-04 18:21:11 +08:00
..
d_tensor [dtensor] fixed readme file name and removed deprecated file (#4162) 2023-07-04 18:21:11 +08:00
__init__.py
colo_parameter.py
colo_tensor.py
comm_spec.py [test] fixed tests failed due to dtensor change (#4082) 2023-07-04 16:05:01 +08:00
compute_spec.py
const.py
dist_spec_mgr.py
distspec.py
op_wrapper.py
param_op_hook.py [nfc]fix typo colossalai/pipeline tensor nn (#3899) 2023-06-06 14:07:36 +08:00
process_group.py [nfc]fix typo colossalai/pipeline tensor nn (#3899) 2023-06-06 14:07:36 +08:00
shape_consistency.py [test] fixed tests failed due to dtensor change (#4082) 2023-07-04 16:05:01 +08:00
sharding_spec.py [test] fixed tests failed due to dtensor change (#4082) 2023-07-04 16:05:01 +08:00
tensor_spec.py
utils.py [nfc]fix typo colossalai/pipeline tensor nn (#3899) 2023-06-06 14:07:36 +08:00