Jiarui Fang 21dc54e019 [zero] memtracer to record cuda memory usage of model data and overall system (#395)

2022-03-14 22:05:30 +08:00

..

__init__.py

move async memory to an individual directory (#345)

2022-03-11 15:50:28 +08:00

allocator.py

[zero] memtracer to record cuda memory usage of model data and overall system (#395)

2022-03-14 22:05:30 +08:00

async_memtracer.py

[zero] memtracer to record cuda memory usage of model data and overall system (#395)

2022-03-14 22:05:30 +08:00

commons.py

[zero] memtracer to record cuda memory usage of model data and overall system (#395)

2022-03-14 22:05:30 +08:00

memstats_collector.py

[zero] memtracer to record cuda memory usage of model data and overall system (#395)

2022-03-14 22:05:30 +08:00

model_data_memtracer.py

[zero] memtracer to record cuda memory usage of model data and overall system (#395)

2022-03-14 22:05:30 +08:00

test_async_memtracer.py

move async memory to an individual directory (#345)

2022-03-11 15:50:28 +08:00

test_memstats_collector.py

[zero] memtracer to record cuda memory usage of model data and overall system (#395)

2022-03-14 22:05:30 +08:00