/
githubmirror
/
ColossalAI
Обзор
Документация
Войти
/
githubmirror
/
ColossalAI
Код
Запросы
0
Пакеты
0
Релизы
0
Аналитика
Безопасность
v0.4.2
ColossalAI
/
tests
/
test_moe
/
..
moe_utils.py
[MoE/ZeRO] Moe refactor with zero refactor (#5821)
2 года назад
test_deepseek_layer.py
[shardformer] DeepseekMoE support (#5871)
2 года назад
test_grad_handler.py
[MoE/ZeRO] Moe refactor with zero refactor (#5821)
2 года назад
test_kernel.py
[MoE/ZeRO] Moe refactor with zero refactor (#5821)
2 года назад
test_mixtral_layer.py
[MoE/ZeRO] Moe refactor with zero refactor (#5821)
2 года назад
test_moe_checkpoint.py
[shardformer] DeepseekMoE support (#5871)
2 года назад
test_moe_ep_tp.py
[MoE/ZeRO] Moe refactor with zero refactor (#5821)
2 года назад
test_moe_group.py
[MoE/ZeRO] Moe refactor with zero refactor (#5821)
2 года назад
test_moe_hybrid_zero.py
[MoE/ZeRO] Moe refactor with zero refactor (#5821)
2 года назад
test_moe_load_balance.py
[MoE/ZeRO] Moe refactor with zero refactor (#5821)
2 года назад
test_moe_zero_fwd_bwd_optim.py
[MoE/ZeRO] Moe refactor with zero refactor (#5821)
2 года назад