Jiarui Fang 193dc8dacb [refactor] refactor the memory utils (#715)

2022-04-11 16:47:57 +08:00

..

__init__.py

[zero] initialize a stateful tensor manager (#614)

2022-04-06 16:18:49 +08:00

async_memtracer.py

[refactor] refactor the memory utils (#715)

2022-04-11 16:47:57 +08:00

memstats_collector.py

[refactor] refactor the memory utils (#715)

2022-04-11 16:47:57 +08:00

model_data_memtracer.py

[hotfix] fix a bug in model data stats tracing (#655)

2022-04-03 21:48:06 +08:00