github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-10-12 18:33:53 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
5bd80b7dd1b55365cd4860a476dba51a99ff6148
ColossalAI/colossalai/utils
History
Ziyue Jiang d73c2b1d79 [Tensor] fix init context (#931) 
* change torch.Parameter to ColoParameter

* fix post assignment for init context

* polish

* polish
2022-05-11 15:48:12 +08:00
..
data_sampler
Refactored docstring to google style
2022-03-29 17:17:47 +08:00
model
[Tensor] fix init context (#931)
2022-05-11 15:48:12 +08:00
multi_tensor_apply
Refactored docstring to google style
2022-03-29 17:17:47 +08:00
profiler
[utils] refactor profiler (#837)
2022-04-24 17:03:59 +08:00
tensor_detector
Refactored docstring to google style
2022-03-29 17:17:47 +08:00
__init__.py
Init Conext supports lazy allocate model memory (#842)
2022-04-22 18:03:35 +08:00
activation_checkpoint.py
[util] fixed activation checkpointing on torch 1.9 (#719)
2022-04-12 09:35:45 +08:00
checkpointing.py
polish checkpoint docstring (#637)
2022-04-02 13:34:33 +08:00
common.py
[zero] refactor memstats collector (#706)
2022-04-11 10:46:08 +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
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