Ziyue Jiang d73c2b1d79 [Tensor] fix init context (#931) ... * change torch.Parameter to ColoParameter * fix post assignment for init context * polish * polish		2022-05-11 15:48:12 +08:00
..
components_to_test	[Tensor] add 1d vocab loss (#918)	2022-05-07 15:49:14 +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	[Tensor] fix init context (#931)	2022-05-11 15:48:12 +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	…