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	…
model	[Tensor] fix init context (#931)	2022-05-11 15:48:12 +08:00
multi_tensor_apply	…
profiler	[utils] refactor profiler (#837)	2022-04-24 17:03:59 +08:00
tensor_detector	…
__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	…
common.py	…
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	…
timer.py	…