ColossalAI/tests/test_tensor/model at e37f3db40c4abce67e42a5cb2493e3fce0e08de3 - ColossalAI - Gitea: Git with a cup of tea at home

github/ColossalAI

mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-07-16 00:14:39 +00:00

Files

History

Jiarui Fang 2e9cbfca12 [Gemini] add unitests to check gemini correctness (#2015 )

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

..

test_gpt2.py

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

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

test_model.py

[utils] remove lazy_memory_allocate from ColoInitContext (#1844 )

2022-11-09 11:50:33 +08:00

test_module_spec.py

[utils] remove lazy_memory_allocate from ColoInitContext (#1844 )

2022-11-09 11:50:33 +08:00