You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
ColossalAI/colossalai/utils
Hongxin Liu b5f9e37c70
[legacy] clean up legacy code (#4743)
1 year ago
..
model fix typo with colossalai/trainer utils zero (#3908) 1 year ago
multi_tensor_apply [setup] support pre-build and jit-build of cuda kernels (#2374) 2 years ago
rank_recorder fix typo with colossalai/trainer utils zero (#3908) 1 year ago
tensor_detector [NFC]fix typo colossalai/auto_parallel nn utils etc. (#3779) 2 years ago
__init__.py [legacy] clean up legacy code (#4743) 1 year ago
common.py [legacy] clean up legacy code (#4743) 1 year ago
cuda.py [legacy] clean up legacy code (#4743) 1 year ago
moe.py [legacy] clean up legacy code (#4743) 1 year ago
timer.py Refactored docstring to google style 3 years ago