Ziyue Jiang 193d629311 update pytest.mark.parametrize in tensor tests (#913)		2022-05-06 11:16:40 +08:00
..
components_to_test	[Tensor] add embedding tp1d row (#904)	2022-04-29 14:10:05 +08:00
test_amp	…
test_comm	…
test_config	…
test_context	…
test_data	[unittest] refactored unit tests for change in dependency (#838)	2022-04-22 15:39:07 +08:00
test_data_pipeline_tensor_parallel	[refactor] moving grad acc logic to engine (#804)	2022-04-19 14:03:21 +08:00
test_engine	[refactor] moving grad acc logic to engine (#804)	2022-04-19 14:03:21 +08:00
test_gemini	[gemini] accelerate adjust_layout() (#878)	2022-04-26 18:08:31 +08:00
test_layers	…
test_moe	…
test_optimizer	…
test_tensor	update pytest.mark.parametrize in tensor tests (#913)	2022-05-06 11:16:40 +08:00
test_trainer	…
test_utils	[gemini] add GeminiMemoryManger (#832)	2022-04-24 13:08:48 +08:00
test_zero	[gemini] add GeminiMemoryManger (#832)	2022-04-24 13:08:48 +08:00
__init__.py	…