1
0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-05-08 16:38:15 +00:00
ColossalAI/colossalai/utils
HELSON f6178728a0
[gemini] fix init bugs for modules ()
* [gemini] fix init bugs for modules

* fix bugs
2022-11-30 17:06:10 +08:00
..
checkpoint [hotfix] fix a running error in test_colo_checkpoint.py () 2022-07-29 15:58:06 +08:00
checkpoint_io [CheckpointIO] a uniform checkpoint I/O module () 2022-11-08 15:15:13 +08:00
data_sampler Refactored docstring to google style 2022-03-29 17:17:47 +08:00
model [gemini] fix init bugs for modules () 2022-11-30 17:06:10 +08:00
multi_tensor_apply [kernel] move all symlinks of kernel to colossalai._C () 2022-11-17 13:42:33 +08:00
profiler [Gemini] clean no used MemTraceOp () 2022-11-17 13:41:54 +08:00
rank_recorder [pipeline/rank_recorder] fix bug when process data before backward | add a tool for multiple ranks debug () 2022-10-09 17:32:57 +08:00
tensor_detector [NFC] polish utils/tensor_detector/__init__.py code style () 2022-09-08 22:11:04 +08:00
__init__.py [refactory] add nn.parallel module () 2022-06-06 15:34:41 +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 [kernel] move all symlinks of kernel to colossalai._C () 2022-11-17 13:42:33 +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 Refactored docstring to google style 2022-03-29 17:17:47 +08:00
timer.py Refactored docstring to google style 2022-03-29 17:17:47 +08:00