Neil Zhu 4403a21d4b [Metax] refactor cutlass moe and optimize flash attention (#5361) ...

* [Metax] refactor moe and flash attention backend
---------

Co-authored-by: zhangchenyi_dl <16219492+zhangchenyidl@user.noreply.gitee.com>

2025-12-10 17:15:17 +08:00

..

apply_rope_qkv.cu

[Metax] refactor cutlass moe and optimize flash attention (#5361)

2025-12-10 17:15:17 +08:00

cache_kv_with_rope.cu

[Metax] refactor cutlass moe and optimize flash attention (#5361)

2025-12-10 17:15:17 +08:00

fused_moe_gemm_kernels.h

[Metax] refactor cutlass moe and optimize flash attention (#5361)

2025-12-10 17:15:17 +08:00

fused_moe_helper.h

[Metax] refactor cutlass moe and optimize flash attention (#5361)

2025-12-10 17:15:17 +08:00

fused_moe_imp_op.h

[Metax] refactor cutlass moe and optimize flash attention (#5361)

2025-12-10 17:15:17 +08:00

fused_moe_op.h

[Metax] refactor cutlass moe and optimize flash attention (#5361)

2025-12-10 17:15:17 +08:00

fused_moe.cu

[Metax] refactor cutlass moe and optimize flash attention (#5361)

2025-12-10 17:15:17 +08:00

moe_dispatch.cu

[Metax] refactor cutlass moe and optimize flash attention (#5361)

2025-12-10 17:15:17 +08:00

moe_ffn.cu

[Metax] refactor cutlass moe and optimize flash attention (#5361)

2025-12-10 17:15:17 +08:00

moe_reduce.cu

[Metax] refactor cutlass moe and optimize flash attention (#5361)

2025-12-10 17:15:17 +08:00