ColossalAI/colossalai/gemini/ophooks
ver217 823f3b9cf4
[doc] add deepspeed citation and copyright (#2996)
* [doc] add deepspeed citation and copyright

* [doc] add deepspeed citation and copyright

* [doc] add deepspeed citation and copyright
2023-03-04 20:08:11 +08:00
..
__init__.py [Gemini] clean no used MemTraceOp (#1970) 2022-11-17 13:41:54 +08:00
_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 [hotfix] remove potiential circle import (#1307) 2022-07-14 13:44:26 +08:00
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 [doc] add deepspeed citation and copyright (#2996) 2023-03-04 20:08:11 +08:00