..
components_to_test
[Gemini] more rigorous unit tests for run_fwd_bwd ( #2034 )
2022-11-29 09:26:06 +08:00
test_amp
[amp] add torch amp test ( #1860 )
2022-11-10 16:40:26 +08:00
test_auto_parallel
[autoparallel] add split handler ( #2032 )
2022-11-29 11:03:51 +08:00
test_comm
[communication] add p2p_v2.py to support communication with List[Any] ( #1407 )
2022-08-09 11:40:04 +08:00
test_config
[pipeline] refactor the pipeline module ( #1087 )
2022-06-10 11:27:38 +08:00
test_context
[test] refactored with the new rerun decorator ( #763 )
2022-04-15 00:33:04 +08:00
test_data
[unittest] refactored unit tests for change in dependency ( #838 )
2022-04-22 15:39:07 +08:00
test_data_pipeline_tensor_parallel
[engin/schedule] use p2p_v2 to recontruct pipeline_schedule ( #1408 )
2022-08-12 11:33:26 +08:00
test_ddp
[zero] add chunk init function for users ( #1729 )
2022-10-18 16:31:22 +08:00
test_device
[autoparallel] Add alpha beta ( #1973 )
2022-11-17 16:01:14 +08:00
test_engine
[hotfix] remove potiential circle import ( #1307 )
2022-07-14 13:44:26 +08:00
test_fx
[fx]Split partition with DAG information ( #2025 )
2022-11-25 17:42:48 +08:00
test_gemini
[hotfix] add bert test for gemini fwd bwd ( #2035 )
2022-11-29 11:19:52 +08:00
test_layers
[inference] overlap comm and compute in Linear1D_Row when stream_chunk_num > 1 ( #1876 )
2022-11-10 17:36:42 +08:00
test_moe
[moe] initialize MoE groups by ProcessGroup ( #1640 )
2022-09-23 17:20:41 +08:00
test_ops
[FAW] export FAW in _ops ( #1438 )
2022-08-11 13:43:24 +08:00
test_optimizer
[kernel] move all symlinks of kernel to `colossalai._C` ( #1971 )
2022-11-17 13:42:33 +08:00
test_pipeline
[rpc] split with dag ( #2028 )
2022-11-29 11:36:28 +08:00
test_tensor
[Gemini] add unitests to check gemini correctness ( #2015 )
2022-11-24 16:51:45 +08:00
test_trainer
[pipeline] refactor the pipeline module ( #1087 )
2022-06-10 11:27:38 +08:00
test_utils
updated flash attention api
2022-11-15 15:25:39 +08:00
test_zero
[Gemini] add bert for MemtracerWrapper unintests ( #1982 )
2022-11-18 14:58:28 +08:00
__init__.py
[zero] Update sharded model v2 using sharded param v2 ( #323 )
2022-03-11 15:50:28 +08:00