ColossalAI/colossalai/legacy/tensor
Rocky Duan cbe34c557c
Fix ColoTensorSpec for py11 (#5440)
2024-03-26 15:56:49 +08:00
..
__init__.py [misc] update pre-commit and run all files (#4752) 2023-09-19 14:20:26 +08:00
compute_spec.py [misc] update pre-commit and run all files (#4752) 2023-09-19 14:20:26 +08:00
const.py [misc] update pre-commit and run all files (#4752) 2023-09-19 14:20:26 +08:00
dist_spec_mgr.py [misc] update pre-commit and run all files (#4752) 2023-09-19 14:20:26 +08:00
distspec.py [misc] update pre-commit and run all files (#4752) 2023-09-19 14:20:26 +08:00
op_wrapper.py [legacy] clean up legacy code (#4743) 2023-09-18 16:31:06 +08:00
process_group.py [hotfix] fix torch 2.0 compatibility (#4936) 2023-10-18 11:05:25 +08:00
tensor_spec.py Fix ColoTensorSpec for py11 (#5440) 2024-03-26 15:56:49 +08:00