ColossalAI/colossalai/gemini/ophooks
Jiarui Fang b3b89865e2
[Gemini] ParamOpHook -> ColoParamOpHook (#2080)
2022-12-05 17:11:06 +08:00
..
__init__.py
_shard_grad_ophook.py [Gemini] rename hooks related to runtime mem tracer (#2076) 2022-12-05 15:00:03 +08:00
_shard_param_ophook.py
runtime_mem_tracer_hook.py [Gemini] ParamOpHook -> ColoParamOpHook (#2080) 2022-12-05 17:11:06 +08:00
utils.py