You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
ColossalAI/tests
LuGY cbac782254
[zero]fix zero ckptIO with offload (#4529)
1 year ago
..
components_to_test
kit [gemini] improve compatibility and add static placement policy (#4479) 1 year ago
test_amp
test_analyzer
test_auto_parallel
test_autochunk
test_booster [gemini] improve compatibility and add static placement policy (#4479) 1 year ago
test_checkpoint_io [zero]fix zero ckptIO with offload (#4529) 1 year ago
test_cluster
test_comm
test_config
test_context
test_data
test_data_pipeline_tensor_parallel [gemini] improve compatibility and add static placement policy (#4479) 1 year ago
test_device
test_engine
test_fx
test_kernels
test_layers
test_lazy
test_moe
test_optimizer
test_pipeline [gemini] improve compatibility and add static placement policy (#4479) 1 year ago
test_shardformer [gemini] improve compatibility and add static placement policy (#4479) 1 year ago
test_tensor [gemini] improve compatibility and add static placement policy (#4479) 1 year ago
test_trainer
test_utils [gemini] improve compatibility and add static placement policy (#4479) 1 year ago
test_zero [zero]fix zero ckptIO with offload (#4529) 1 year ago
__init__.py