You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 
 
 
 
Yichen Zhang f1a09acba4
[AutoParallel] 3D parallel on MLP with PIR (#64369)
1 day ago
..
CMakeLists.txt [DistDialect] Add pir nd_mesh reshard function (#64223) 4 days ago
pir_reshard_nd_mesh.py [AutoParallel] 3D parallel on MLP with PIR (#64369) 1 day ago
pir_reshard_nd_mesh_cross_mesh.py [DistDialect] Add pir nd_mesh reshard function (#64223) 4 days ago
semi_auto_llama.py [AutoParallel] fix hang when vpp enabled (#62727) 2 months ago
semi_auto_llama_acc_align.py [AutoParallel] fix hang when vpp enabled (#62727) 2 months ago
semi_auto_llama_pp_gradmerge.py [AutoParallel] fix hang when vpp enabled (#62727) 2 months ago
semi_auto_load_state_dict.py Set DistModel state_dict keys to structure_names (#60478) 4 months ago
semi_auto_parallel_2d_global_mesh_reshard.py [AutoParallel] Add Global mesh and sub mesh reshard function (#61796) 2 months ago
semi_auto_parallel_3d_global_mesh_reshard.py support 3d mesh calculation (#62356) 2 months ago
semi_auto_parallel_c_cross_entropy_dp.py [Auto Parallel] fix the bug of parallel cross_entropy unit test (#59961) 5 months ago
semi_auto_parallel_c_cross_entropy_hybrid.py [Auto Parallel] fix the bug of parallel cross_entropy unit test (#59961) 5 months ago
semi_auto_parallel_c_cross_entropy_mp.py [Auto Parallel] fix the bug of parallel cross_entropy unit test (#59961) 5 months ago
semi_auto_parallel_cross_mesh_reshard.py [auto parallel] CrossMeshReshard for: p2r, p2s, r2p, r2s, s2p, s2r, s2s. (#59758) 5 months ago
semi_auto_parallel_for_fused_linear_param_grad_add.py fused_linear_param_grad infer spmd (#60144) 5 months ago
semi_auto_parallel_for_llama_decoder_dp_mp.py Auto recompute enable (#60120) 4 months ago
semi_auto_parallel_global_input.py Fix Registe Register, etc (#63429) 1 month ago
semi_auto_parallel_llama_model.py [AutoParallel] fix hang when vpp enabled (#62727) 2 months ago
semi_auto_parallel_multi_inputs.py Fix Registe Register, etc (#63429) 1 month ago
semi_auto_parallel_mutual_load_between_dynamic_and_static.py [auto parallel] Add shard_dataloader api (#60728) 4 months ago
semi_auto_parallel_nd_cross_mesh_reshard.py [auto parallel] Nd reshard cross mesh (#59777) 5 months ago
semi_auto_parallel_sharding_stage_1.py make sharding dynamic to static (#62230) 2 months ago
semi_auto_parallel_sharding_stage_2.py support sharding stage 2 (#62486) 2 months ago
semi_auto_parallel_sharding_stage_3.py make sharding dynamic to static (#62230) 2 months ago
semi_auto_parallel_simple_net_dp_mp.py Set DistModel state_dict keys to structure_names (#60478) 4 months ago
semi_auto_parallel_simple_net_dp_mp_pp.py Fix lable label (#60951) 4 months ago
semi_auto_parallel_simple_net_sp.py [AutoParallel] Fix pipeline parallel get none grad in non-computatio rank. (#60214) 4 months ago
semi_auto_save_state_dict.py Set DistModel state_dict keys to structure_names (#60478) 4 months ago
test_cross_mesh_reshard.py Fix lable label (#60951) 4 months ago
test_global_mesh_reshard.py [AutoParallel] Add Global mesh and sub mesh reshard function (#61796) 2 months ago
test_pir_reshard_nd_mesh_func.py [DistDialect] Add pir nd_mesh reshard function (#64223) 4 days ago
test_save_load_state_dict.py Set DistModel state_dict keys to structure_names (#60478) 4 months ago
test_semi_auto_parallel_c_cross_entropy.py 【PIR OpTest Fix No.38】 fix test_semi_auto_parallel_c_cross_entropy (#59893) 4 months ago
test_semi_auto_parallel_global_input.py [AutoParallel] add global_input unittest (#62115) 2 months ago
test_semi_auto_parallel_hybrid_sharding_strategy.py support sharding stage 2 (#62486) 2 months ago
test_semi_auto_parallel_hybrid_strategy.py [Auto Parallel] Add target path for semi-auto and remove llama uts into a new file (#60273) 4 months ago
test_semi_auto_parallel_llama_model.py reconstruct llama ci cases (#60637) 4 months ago
test_semi_auto_parallel_llama_model_amp.py 【AutoParallism】Support semi auto amp (#61221) 3 months ago
test_semi_auto_parallel_llama_model_pir.py [AutoParallel-PIR] 3D HybridParallel Demo & Unitest (#64143) 4 days ago
test_semi_auto_parallel_llama_model_vpp.py [AutoParallel] fix hang when vpp enabled (#62727) 2 months ago
test_semi_auto_parallel_multi_inputs.py [AutoParallel] shard_dataloader support list inputs (#62229) 2 months ago
testslist.csv [DistDialect] Add pir nd_mesh reshard function (#64223) 4 days ago