ColossalAI/tests/test_gemini
Jiarui Fang 96134e7be3
[hotfix] add bert test for gemini fwd bwd (#2035)
2022-11-29 11:19:52 +08:00
..
update [hotfix] add bert test for gemini fwd bwd (#2035) 2022-11-29 11:19:52 +08:00
test_gemini_manager.py
test_mem_tracer.py [Gemini] more rigorous unit tests for run_fwd_bwd (#2034) 2022-11-29 09:26:06 +08:00
test_param_op.py [Gemini] add unitests to check gemini correctness (#2015) 2022-11-24 16:51:45 +08:00
test_param_tracer.py [Gemini] paramWrapper paramTracerHook unitest (#2030) 2022-11-26 13:30:24 +08:00
test_stateful_tensor_mgr.py