Add DSA MLS sparse prefill dispatch

a1eef562 · shenzhe · zhanghj2 · 4e0bdf6e · a1eef562 · a1eef562
Commit a1eef562 authored Jun 04, 2026 by shenzhe Committed by zhanghj2 Jun 06, 2026
20 changed files
--- a/csrc/gfx93/prefill/sparse/dsa_mls/legacy/include/mla/mla_pv_gemm_prefetch_k_tile16x32.h
+++ b/csrc/gfx93/prefill/sparse/dsa_mls/legacy/include/mla/mla_pv_gemm_prefetch_k_tile16x32.h
--- a/csrc/gfx93/prefill/sparse/dsa_mls/legacy/include/mla/mla_pv_gemm_utils.h
+++ b/csrc/gfx93/prefill/sparse/dsa_mls/legacy/include/mla/mla_pv_gemm_utils.h
--- a/csrc/gfx93/prefill/sparse/dsa_mls/legacy/include/mla/mla_pv_gemm_utils_tile16x32.h
+++ b/csrc/gfx93/prefill/sparse/dsa_mls/legacy/include/mla/mla_pv_gemm_utils_tile16x32.h
--- a/csrc/gfx93/prefill/sparse/dsa_mls/legacy/include/mla/mla_qk_gemm_prefetch_v.h
+++ b/csrc/gfx93/prefill/sparse/dsa_mls/legacy/include/mla/mla_qk_gemm_prefetch_v.h
--- a/csrc/gfx93/prefill/sparse/dsa_mls/legacy/include/mla/mla_qk_gemm_prefetch_v_qinlds.h
+++ b/csrc/gfx93/prefill/sparse/dsa_mls/legacy/include/mla/mla_qk_gemm_prefetch_v_qinlds.h
--- a/csrc/gfx93/prefill/sparse/dsa_mls/legacy/include/mla/mla_qk_gemm_prefetch_v_tile16x32.h
+++ b/csrc/gfx93/prefill/sparse/dsa_mls/legacy/include/mla/mla_qk_gemm_prefetch_v_tile16x32.h
--- a/csrc/gfx93/prefill/sparse/dsa_mls/legacy/include/mla/mla_qk_gemm_utils.h
+++ b/csrc/gfx93/prefill/sparse/dsa_mls/legacy/include/mla/mla_qk_gemm_utils.h
--- a/csrc/gfx93/prefill/sparse/dsa_mls/legacy/include/mla/mla_qk_gemm_utils_tile16x32.h
+++ b/csrc/gfx93/prefill/sparse/dsa_mls/legacy/include/mla/mla_qk_gemm_utils_tile16x32.h
--- a/csrc/gfx93/prefill/sparse/dsa_mls/legacy/include/mla/mla_softmax.h
+++ b/csrc/gfx93/prefill/sparse/dsa_mls/legacy/include/mla/mla_softmax.h
--- a/csrc/gfx93/prefill/sparse/dsa_mls/legacy/include/mla/mla_softmax_tile16x32.h
+++ b/csrc/gfx93/prefill/sparse/dsa_mls/legacy/include/mla/mla_softmax_tile16x32.h
--- a/csrc/gfx93/prefill/sparse/dsa_mls/legacy/include/numeric_types.h
+++ b/csrc/gfx93/prefill/sparse/dsa_mls/legacy/include/numeric_types.h
--- a/csrc/gfx93/prefill/sparse/dsa_mls/legacy/include/philox.cuh
+++ b/csrc/gfx93/prefill/sparse/dsa_mls/legacy/include/philox.cuh
--- a/csrc/gfx93/prefill/sparse/dsa_mls/legacy/include/splitkv.h
+++ b/csrc/gfx93/prefill/sparse/dsa_mls/legacy/include/splitkv.h
--- a/csrc/gfx93/prefill/sparse/dsa_mls/legacy/include/static_switch.h
+++ b/csrc/gfx93/prefill/sparse/dsa_mls/legacy/include/static_switch.h
--- a/csrc/gfx93/prefill/sparse/dsa_mls/legacy/include/wait.h
+++ b/csrc/gfx93/prefill/sparse/dsa_mls/legacy/include/wait.h
--- a/csrc/gfx93/prefill/sparse/dsa_mls/legacy/src/flash_fwd_b16_mla.h
+++ b/csrc/gfx93/prefill/sparse/dsa_mls/legacy/src/flash_fwd_b16_mla.h
--- a/csrc/gfx93/prefill/sparse/dsa_mls/legacy/src/flash_fwd_launch_template_mla.h
+++ b/csrc/gfx93/prefill/sparse/dsa_mls/legacy/src/flash_fwd_launch_template_mla.h
--- a/csrc/gfx93/prefill/sparse/dsa_mls/legacy/src/flash_fwd_launch_template_pa.h
+++ b/csrc/gfx93/prefill/sparse/dsa_mls/legacy/src/flash_fwd_launch_template_pa.h
--- a/csrc/gfx93/prefill/sparse/dsa_mls/legacy/src/flash_fwd_reduce.h
+++ b/csrc/gfx93/prefill/sparse/dsa_mls/legacy/src/flash_fwd_reduce.h
--- a/csrc/gfx93/prefill/sparse/fwd.cu
+++ b/csrc/gfx93/prefill/sparse/fwd.cu