ColossalAI/tests
YuliangLiu0306 9c9246c0d9
[device] alpha beta profiler (#2311)
* [device] alpha beta profiler

* add usage

* fix variable name
2023-01-05 16:39:55 +08:00
..
components_to_test [testing] add beit model for unit testings (#2196) 2022-12-26 17:35:36 +08:00
test_amp [amp] add gradient clipping for unit tests (#2283) 2023-01-04 11:59:56 +08:00
test_auto_parallel [autoparallel] fix spelling error (#2270) 2023-01-03 16:13:00 +08:00
test_comm
test_config
test_context
test_data
test_data_pipeline_tensor_parallel
test_ddp
test_device [device] alpha beta profiler (#2311) 2023-01-05 16:39:55 +08:00
test_engine
test_fx
test_gemini [testing] add beit model for unit testings (#2196) 2022-12-26 17:35:36 +08:00
test_layers improved allgather & reducescatter for 3d 2023-01-03 17:46:08 +08:00
test_moe
test_ops
test_optimizer [builder] reconfig op_builder for pypi install (#2314) 2023-01-04 16:32:32 +08:00
test_pipeline
test_tensor [testing] add beit model for unit testings (#2196) 2022-12-26 17:35:36 +08:00
test_trainer
test_utils updated attention kernel (#2133) 2022-12-16 10:54:03 +08:00
test_zero [testing] add beit model for unit testings (#2196) 2022-12-26 17:35:36 +08:00
__init__.py