ColossalAI/tests/test_layers/test_1d/checks_1d
Jiarui Fang 986f8cbaa7
[inference] overlap comm and compute in Linear1D_Row when stream_chunk_num > 1 (#1876)
2022-11-10 17:36:42 +08:00
..
__init__.py added CI for unit testing (#69) 2021-12-16 10:32:08 +08:00
check_layer_1d.py [inference] overlap comm and compute in Linear1D_Row when stream_chunk_num > 1 (#1876) 2022-11-10 17:36:42 +08:00
common.py moved env variables to global variables; (#215) 2022-02-15 11:31:13 +08:00