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
process_group.py
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