ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-05 14:11:29 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
c0adab485020b83f324d2efdcac2c997e19443eb
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