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/fwd/pv_gemm_utils.h
+++ b/csrc/gfx93/prefill/sparse/dsa_mls/legacy/include/fwd/pv_gemm_utils.h
--- a/csrc/gfx93/prefill/sparse/dsa_mls/legacy/include/fwd/qk_gemm_prefetch_v.h
+++ b/csrc/gfx93/prefill/sparse/dsa_mls/legacy/include/fwd/qk_gemm_prefetch_v.h
--- a/csrc/gfx93/prefill/sparse/dsa_mls/legacy/include/fwd/qk_gemm_prefetch_v_headdim128.h
+++ b/csrc/gfx93/prefill/sparse/dsa_mls/legacy/include/fwd/qk_gemm_prefetch_v_headdim128.h
--- a/csrc/gfx93/prefill/sparse/dsa_mls/legacy/include/fwd/qk_gemm_utils.h
+++ b/csrc/gfx93/prefill/sparse/dsa_mls/legacy/include/fwd/qk_gemm_utils.h
--- a/csrc/gfx93/prefill/sparse/dsa_mls/legacy/include/fwd/softmax.h
+++ b/csrc/gfx93/prefill/sparse/dsa_mls/legacy/include/fwd/softmax.h
--- a/csrc/gfx93/prefill/sparse/dsa_mls/legacy/include/fwd/utils.h
+++ b/csrc/gfx93/prefill/sparse/dsa_mls/legacy/include/fwd/utils.h
--- a/csrc/gfx93/prefill/sparse/dsa_mls/legacy/include/intrinsic.h
+++ b/csrc/gfx93/prefill/sparse/dsa_mls/legacy/include/intrinsic.h
--- a/csrc/gfx93/prefill/sparse/dsa_mls/legacy/include/intrinsic_mls_ds.h
+++ b/csrc/gfx93/prefill/sparse/dsa_mls/legacy/include/intrinsic_mls_ds.h
--- a/csrc/gfx93/prefill/sparse/dsa_mls/legacy/include/intrinsic_mls_ds_b8.h
+++ b/csrc/gfx93/prefill/sparse/dsa_mls/legacy/include/intrinsic_mls_ds_b8.h
--- a/csrc/gfx93/prefill/sparse/dsa_mls/legacy/include/kernel_traits.h
+++ b/csrc/gfx93/prefill/sparse/dsa_mls/legacy/include/kernel_traits.h
--- a/csrc/gfx93/prefill/sparse/dsa_mls/legacy/include/kvcache/gfx938/kvcache_epilogue_gfx938.h
+++ b/csrc/gfx93/prefill/sparse/dsa_mls/legacy/include/kvcache/gfx938/kvcache_epilogue_gfx938.h
--- a/csrc/gfx93/prefill/sparse/dsa_mls/legacy/include/kvcache/gfx938/kvcache_pv_gemm_prefetch_k_gfx938.h
+++ b/csrc/gfx93/prefill/sparse/dsa_mls/legacy/include/kvcache/gfx938/kvcache_pv_gemm_prefetch_k_gfx938.h
--- a/csrc/gfx93/prefill/sparse/dsa_mls/legacy/include/kvcache/gfx938/kvcache_pv_gemm_utils_gfx938.h
+++ b/csrc/gfx93/prefill/sparse/dsa_mls/legacy/include/kvcache/gfx938/kvcache_pv_gemm_utils_gfx938.h
--- a/csrc/gfx93/prefill/sparse/dsa_mls/legacy/include/kvcache/gfx938/kvcache_qk_gemm_prefetch_v_gfx938.h
+++ b/csrc/gfx93/prefill/sparse/dsa_mls/legacy/include/kvcache/gfx938/kvcache_qk_gemm_prefetch_v_gfx938.h
--- a/csrc/gfx93/prefill/sparse/dsa_mls/legacy/include/kvcache/gfx938/kvcache_qk_gemm_utils_gfx938.h
+++ b/csrc/gfx93/prefill/sparse/dsa_mls/legacy/include/kvcache/gfx938/kvcache_qk_gemm_utils_gfx938.h
--- a/csrc/gfx93/prefill/sparse/dsa_mls/legacy/include/kvcache/gfx938/kvcache_softmax_gfx938.h
+++ b/csrc/gfx93/prefill/sparse/dsa_mls/legacy/include/kvcache/gfx938/kvcache_softmax_gfx938.h
--- a/csrc/gfx93/prefill/sparse/dsa_mls/legacy/include/kvcache/int8_kvcache_acco_reduce.h
+++ b/csrc/gfx93/prefill/sparse/dsa_mls/legacy/include/kvcache/int8_kvcache_acco_reduce.h
--- a/csrc/gfx93/prefill/sparse/dsa_mls/legacy/include/kvcache/int8_kvcache_pv_gemm_prefetch_k.h
+++ b/csrc/gfx93/prefill/sparse/dsa_mls/legacy/include/kvcache/int8_kvcache_pv_gemm_prefetch_k.h
--- a/csrc/gfx93/prefill/sparse/dsa_mls/legacy/include/kvcache/int8_kvcache_pv_gemm_prefetch_k_3stage.h
+++ b/csrc/gfx93/prefill/sparse/dsa_mls/legacy/include/kvcache/int8_kvcache_pv_gemm_prefetch_k_3stage.h
--- a/csrc/gfx93/prefill/sparse/dsa_mls/legacy/include/kvcache/int8_kvcache_pv_gemm_utils.h
+++ b/csrc/gfx93/prefill/sparse/dsa_mls/legacy/include/kvcache/int8_kvcache_pv_gemm_utils.h