ColossalAI/tests/components_to_test/utils
Jiarui Fang 28aa9a4294
[Gemini] more rigorous unit tests for run_fwd_bwd (#2034)
2022-11-29 09:26:06 +08:00
..
__init__.py [Gemini] add unitests to check gemini correctness (#2015) 2022-11-24 16:51:45 +08:00
dummy_data_generator.py [unit test] Refactored test cases with component func (#339) 2022-03-11 15:50:28 +08:00
executor.py [Gemini] more rigorous unit tests for run_fwd_bwd (#2034) 2022-11-29 09:26:06 +08:00