ColossalAI/colossalai/gemini/ophooks
Jiarui Fang 2938edf446
[Gemini] update the non model data record method in runtime memory tracer (#2128)
2022-12-13 17:11:31 +08:00
..
__init__.py
_shard_grad_ophook.py
_shard_param_ophook.py
runtime_mem_tracer_hook.py [Gemini] update the non model data record method in runtime memory tracer (#2128) 2022-12-13 17:11:31 +08:00
utils.py