ColossalAI/colossalai/legacy/tensor
Hongxin Liu 1f5d2e8062
[hotfix] fix torch 2.0 compatibility (#4936)
* [hotfix] fix launch

* [test] fix test gemini optim

* [shardformer] fix vit
2023-10-18 11:05:25 +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 [misc] update pre-commit and run all files (#4752) 2023-09-19 14:20:26 +08:00