githubmirror/ColossalAI | Gitverse

/

/

Обзор ДокументацияВойти

/

/

Аналитика

Безопасность

tests/test_moe/

[moe] init mixtral impl

3 года назад

test_grad_handler.py

[npu] change device to accelerator api (#5239)

3 года назад

[npu] change device to accelerator api (#5239)

3 года назад

test_moe_checkpoint.py

[moe] fix tests

3 года назад

test_moe_ep_tp.py

[npu] change device to accelerator api (#5239)

3 года назад

test_moe_group.py

[npu] change device to accelerator api (#5239)

3 года назад

test_moe_hybrid_zero.py

[moe] support optimizer checkpoint (#5015)

3 года назад

test_moe_load_balance.py

[moe] support optimizer checkpoint (#5015)

3 года назад

test_moe_router.py

[moe] fix tests

3 года назад

test_moe_zero_fwd_bwd.py

[moe] init mixtral impl

3 года назад

test_moe_zero_optim.py

[moe] init mixtral impl

3 года назад