githubmirror/ColossalAI | Gitverse

/

/

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

/

/

Аналитика

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

tests/test_moe/

[moe] init mixtral impl

3 года назад

test_grad_handler.py

[misc] refactor launch API and tensor constructor (#5666)

2 года назад

[misc] refactor launch API and tensor constructor (#5666)

2 года назад

test_moe_checkpoint.py

[hotfix] set return_outputs=False in examples and polish code (#5404)

2 года назад

test_moe_ep_tp.py

[misc] refactor launch API and tensor constructor (#5666)

2 года назад

test_moe_group.py

[misc] refactor launch API and tensor constructor (#5666)

2 года назад

test_moe_hybrid_zero.py

[misc] refactor launch API and tensor constructor (#5666)

2 года назад

test_moe_load_balance.py

[misc] refactor launch API and tensor constructor (#5666)

2 года назад

test_moe_router.py

[moe] fix tests

3 года назад

test_moe_zero_fwd_bwd.py

[misc] refactor launch API and tensor constructor (#5666)

2 года назад

test_moe_zero_optim.py

[misc] refactor launch API and tensor constructor (#5666)

2 года назад