..
components_to_test
[Gemini] add albert in test models. ( #2075 )
2022-12-05 14:09:34 +08:00
test_amp
[amp] add torch amp test ( #1860 )
2022-11-10 16:40:26 +08:00
test_auto_parallel
[autoparallel] add bias addtion function class ( #2098 )
2022-12-08 11:31: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
[Pipeline] Add Topo Class ( #2059 )
2022-12-02 18:13:20 +08:00
test_gemini
[Gemini] gemini use the runtime memory tracer (RMT) ( #2099 )
2022-12-07 23:04:02 +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
[test] align model name with the file name. ( #2045 )
2022-11-30 15:45:26 +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
[Pipeline Middleware] fix data race in Pipeline Scheduler for DAG ( #2087 )
2022-12-08 13:32:27 +08:00
test_tensor
[Gemini] remove static tracer ( #2083 )
2022-12-06 12:53:58 +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] remove GLOBAL_MODEL_DATA_TRACER ( #2091 )
2022-12-06 22:30:16 +08:00
__init__.py
[zero] Update sharded model v2 using sharded param v2 ( #323 )
2022-03-11 15:50:28 +08:00
e4705ba4e2