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

..

fused_moe_api.cpp

[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_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)

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

fused_moegemm_bf16_m32.cpp

[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_fp16_m32.cpp

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

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

fused_moesorting_api.cpp

[CK_TILE] moe sorting ex kernel to support expert > 128 (#1840)

2025-02-11 17:49:17 +08:00