ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-03 13:11:25 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
4cfb24feb67602d38b60a1568492c6313bf25a82
composable_kernel/include/ck_tile/ops/fused_moe
History
carlushuang c0adab4850 [CK_TILE] moe sorting ex kernel to support expert > 128 (#1840) 
* moe sorting ex

* fix bug for race condition

* fix bug and optimze large expert

* fix

* optimize with sub_token_oneshot

* support skip empty tokens for expert sorting

* update moe_sorting

* tidy code
2025-02-11 17:49:17 +08:00
..
kernel
[CK_TILE] moe sorting ex kernel to support expert > 128 (#1840)
2025-02-11 17:49:17 +08:00
pipeline
[CK_TILE] moe sorting ex kernel to support expert > 128 (#1840)
2025-02-11 17:49:17 +08:00