github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-05-03 22:18:23 +00:00
Code Issues Packages Projects Releases Wiki Activity
ac178ca5c1
ColossalAI/colossalai/utils
History
Hongxin Liu ac178ca5c1 [legacy] move builder and registry to legacy ()
2023-09-05 21:53:10 +08:00
..
checkpoint [doc] Fix typo under colossalai and doc() 2023-04-26 11:38:43 +08:00
data_sampler [legacy] move builder and registry to legacy () 2023-09-05 21:53:10 +08:00
model fix typo with colossalai/trainer utils zero () 2023-06-07 16:08:37 +08:00
multi_tensor_apply [setup] support pre-build and jit-build of cuda kernels () 2023-01-06 20:50:26 +08:00
profiler [legacy] move engine to legacy () 2023-09-05 21:53:10 +08:00
rank_recorder fix typo with colossalai/trainer utils zero () 2023-06-07 16:08:37 +08:00
tensor_detector [NFC]fix typo colossalai/auto_parallel nn utils etc. () 2023-05-23 15:28:20 +08:00
__init__.py [test] refactor tests with spawn () 2023-04-06 14:51:35 +08:00
activation_checkpoint.py [utils] Add use_reetrant=False in utils.activation_checkpoint () 2022-08-16 15:39:20 +08:00
checkpointing.py [utils] refactor parallel layers checkpoint and bcast model on loading checkpoint () 2022-09-06 20:18:35 +08:00
common.py [NFC]fix typo colossalai/auto_parallel nn utils etc. () 2023-05-23 15:28:20 +08:00
cuda.py [refactor] refactor the memory utils () 2022-04-11 16:47:57 +08:00
memory.py [gemini] APIs to set cpu memory capacity () 2022-04-19 16:05:22 +08:00
moe.py [doc] Fix typo under colossalai and doc() 2023-04-26 11:38:43 +08:00
timer.py Refactored docstring to google style 2022-03-29 17:17:47 +08:00