| .. |
|
checkpoint
|
[doc] Fix typo under colossalai and doc(#3618)
|
2023-04-26 11:38:43 +08:00 |
|
checkpoint_io
|
[CheckpointIO] a uniform checkpoint I/O module (#1689)
|
2022-11-08 15:15:13 +08:00 |
|
data_sampler
|
fix typo with colossalai/trainer utils zero (#3908)
|
2023-06-07 16:08:37 +08:00 |
|
model
|
fix typo with colossalai/trainer utils zero (#3908)
|
2023-06-07 16:08:37 +08:00 |
|
multi_tensor_apply
|
[setup] support pre-build and jit-build of cuda kernels (#2374)
|
2023-01-06 20:50:26 +08:00 |
|
profiler
|
fix typo with colossalai/trainer utils zero (#3908)
|
2023-06-07 16:08:37 +08:00 |
|
rank_recorder
|
fix typo with colossalai/trainer utils zero (#3908)
|
2023-06-07 16:08:37 +08:00 |
|
tensor_detector
|
[NFC]fix typo colossalai/auto_parallel nn utils etc. (#3779)
|
2023-05-23 15:28:20 +08:00 |
|
__init__.py
|
[test] refactor tests with spawn (#3452)
|
2023-04-06 14:51:35 +08:00 |
|
activation_checkpoint.py
|
[utils] Add use_reetrant=False in utils.activation_checkpoint (#1460)
|
2022-08-16 15:39:20 +08:00 |
|
checkpointing.py
|
[utils] refactor parallel layers checkpoint and bcast model on loading checkpoint (#1548)
|
2022-09-06 20:18:35 +08:00 |
|
common.py
|
[NFC]fix typo colossalai/auto_parallel nn utils etc. (#3779)
|
2023-05-23 15:28:20 +08:00 |
|
cuda.py
|
[refactor] refactor the memory utils (#715)
|
2022-04-11 16:47:57 +08:00 |
|
memory.py
|
[gemini] APIs to set cpu memory capacity (#809)
|
2022-04-19 16:05:22 +08:00 |
|
moe.py
|
[doc] Fix typo under colossalai and doc(#3618)
|
2023-04-26 11:38:43 +08:00 |
|
timer.py
|
…
|
|
a9d1cadc49