Jiarui Fang 867c8c2d3a [zero] low level optim supports ProcessGroup (#2464)		2023-01-13 10:05:58 +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] update binary elementwise handler (#2451)	2023-01-12 09:35:10 +08:00
test_autochunk	update doc	2023-01-10 12:29:09 +08:00
test_comm	…
test_config	…
test_context	…
test_data	…
test_data_pipeline_tensor_parallel	…
test_ddp	…
test_device	[device] find best logical mesh	2023-01-05 17:21:29 +08:00
test_engine	…
test_fx	…
test_gemini	[zero] fix state_dict and load_state_dict for ddp ignored parameters (#2443)	2023-01-11 14:55:41 +08:00
test_layers	improved allgather & reducescatter for 3d	2023-01-03 17:46:08 +08:00
test_moe	…
test_ops	…
test_optimizer	[setup] support pre-build and jit-build of cuda kernels (#2374)	2023-01-06 20:50:26 +08:00
test_pipeline	…
test_tensor	[polish] polish code for get_static_torch_model (#2405)	2023-01-09 17:41:38 +08:00
test_trainer	…
test_utils	updated attention kernel (#2133)	2022-12-16 10:54:03 +08:00
test_zero	[zero] low level optim supports ProcessGroup (#2464)	2023-01-13 10:05:58 +08:00
__init__.py	…