Zihao 6a9158f1fa [Gemini] free and allocate cuda memory by tensor.storage, add grad hook (#2040)

2022-11-30 15:57:45 +08:00

..

update

[test] align model name with the file name. (#2045)

2022-11-30 15:45:26 +08:00

test_gemini_manager.py

[gemini] add GeminiMemoryManger (#832)

2022-04-24 13:08:48 +08:00

test_mem_tracer.py

[test] align model name with the file name. (#2045)

2022-11-30 15:45:26 +08:00

test_param_op.py

[test] align model name with the file name. (#2045)

2022-11-30 15:45:26 +08:00

test_param_tracer.py

[Gemini] free and allocate cuda memory by tensor.storage, add grad hook (#2040)

2022-11-30 15:57:45 +08:00

test_stateful_tensor_mgr.py

[gemini] accelerate adjust_layout() (#878)

2022-04-26 18:08:31 +08:00