composable_kernel

mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-02 04:31:25 +00:00

Files

carlushuang 353a612b44 [CK_TILE] add moe-sorting MP kernel (#1910 )

* 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

* support mp kernel

* hint mp

* remove use less code

* porting to example 15

---------

Co-authored-by: valarLip <340077269@qq.com>

2025-02-25 17:56:55 +08:00

fused_moe_api.cpp

[CK_TILE] add moe-sorting MP kernel (#1910 )

2025-02-25 17:56:55 +08:00

fused_moegemm_api_internal.hpp

[CK_TILE] Fix mock token id, support g1u1/g1u0 through same inline code block (#1808 )

2025-01-16 17:51:10 +08:00

fused_moegemm_api_traits.hpp

[CK_TILE] Fix mock token id, support g1u1/g1u0 through same inline code block (#1808 )

2025-01-16 17:51:10 +08:00

fused_moegemm_api.cpp

[CK_TILE] Fix mock token id, support g1u1/g1u0 through same inline code block (#1808 )