Jiarui Fang 8daf1b4db1 [Gemini] patch for supporting orch.add_ function for ColoTensor (#2003)

2022-11-25 20:06:35 +08:00

..

update

[Gemini] add unitests to check gemini correctness (#2015)

2022-11-24 16:51:45 +08:00

test_gemini_manager.py

[gemini] add GeminiMemoryManger (#832)

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

test_gemini_train.py

[Gemini] patch for supporting orch.add_ function for ColoTensor (#2003)

2022-11-25 20:06:35 +08:00

test_mem_tracer.py

[Gemini] add unitests to check gemini correctness (#2015)

2022-11-24 16:51:45 +08:00

test_param_op.py

[Gemini] add unitests to check gemini correctness (#2015)

2022-11-24 16:51:45 +08:00

test_stateful_tensor_mgr.py

[gemini] accelerate adjust_layout() (#878)

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