ColossalAI/tests/test_tensor/model
Jiarui Fang 1f99205827
[Gemini] remove static tracer (#2083)
2022-12-06 12:53:58 +08:00
..
test_gpt2.py [Gemini] add unitests to check gemini correctness (#2015) 2022-11-24 16:51:45 +08:00
test_model.py [Gemini] remove static tracer (#2083) 2022-12-06 12:53:58 +08:00
test_module_spec.py [utils] remove lazy_memory_allocate from ColoInitContext (#1844) 2022-11-09 11:50:33 +08:00