1
0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-05-08 00:18:25 +00:00
ColossalAI/colossalai/moe
botbw c54c4fcd15
[hotfix] moe hybrid parallelism benchmark & follow-up fix ()
* [example] pass use_fp8_comm flag to all plugins

* [example] add mixtral benchmark

* [moe] refine assertion and check

* [moe] fix mixtral & add more tests

* [moe] consider checking dp * sp group and moe_dp_group

* [mixtral] remove gate tp & add more tests

* [deepseek] fix tp & sp for deepseek

* [mixtral] minor fix

* [deepseek] add deepseek benchmark
2024-09-10 17:30:53 +08:00
..
__init__.py [FP8] rebase main () 2024-08-06 16:29:37 +08:00
_operation.py [hotfix] moe hybrid parallelism benchmark & follow-up fix () 2024-09-10 17:30:53 +08:00