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/test_moe
botbw c54c4fcd15
[hotfix] moe hybrid parallelism benchmark & follow-up fix (#6048)
2 months ago
..
moe_utils.py [hotfix] moe hybrid parallelism benchmark & follow-up fix (#6048) 2 months ago
test_deepseek_layer.py
test_kernel.py
test_mixtral_layer.py
test_moe_checkpoint.py [hotfix] moe hybrid parallelism benchmark & follow-up fix (#6048) 2 months ago
test_moe_ep_tp.py
test_moe_ep_zero.py