空kernel可以编译通过

e2e0225c · zhanghj2 · 48c6dc42 · 48c6dc42 · 48c6dc42 · 48c6dc42
Commit e2e0225c authored Jan 25, 2026 by zhanghj2
20 changed files
--- a/csrc/sm100/prefill/dense/device/fmha_device_bwd.hpp
+++ b/csrc/sm100/prefill/dense/device/fmha_device_bwd.hpp
--- a/csrc/sm100/prefill/dense/fmha_cutlass_bwd_sm100.cu
+++ b/csrc/sm100/prefill/dense/fmha_cutlass_bwd_sm100.cu
--- a/csrc/sm100/prefill/dense/fmha_cutlass_bwd_sm100.cuh
+++ b/csrc/sm100/prefill/dense/fmha_cutlass_bwd_sm100.cuh
--- a/csrc/sm100/prefill/dense/fmha_cutlass_fwd_sm100.cu
+++ b/csrc/sm100/prefill/dense/fmha_cutlass_fwd_sm100.cu
--- a/csrc/sm100/prefill/dense/fmha_cutlass_fwd_sm100.cuh
+++ b/csrc/sm100/prefill/dense/fmha_cutlass_fwd_sm100.cuh
--- a/csrc/sm100/prefill/dense/interface.h
+++ b/csrc/sm100/prefill/dense/interface.h
--- a/csrc/sm100/prefill/dense/kernel/fmha_causal_tile_scheduler.hpp
+++ b/csrc/sm100/prefill/dense/kernel/fmha_causal_tile_scheduler.hpp
--- a/csrc/sm100/prefill/dense/kernel/fmha_kernel_bwd_convert.hpp
+++ b/csrc/sm100/prefill/dense/kernel/fmha_kernel_bwd_convert.hpp
--- a/csrc/sm100/prefill/dense/kernel/fmha_kernel_bwd_sum_OdO.hpp
+++ b/csrc/sm100/prefill/dense/kernel/fmha_kernel_bwd_sum_OdO.hpp
--- a/csrc/sm100/prefill/dense/kernel/fmha_options.hpp
+++ b/csrc/sm100/prefill/dense/kernel/fmha_options.hpp
--- a/csrc/sm100/prefill/dense/kernel/fmha_tile_scheduler.hpp
+++ b/csrc/sm100/prefill/dense/kernel/fmha_tile_scheduler.hpp
--- a/csrc/sm100/prefill/dense/kernel/sm100_fmha_bwd_kernel_tma_warpspecialized.hpp
+++ b/csrc/sm100/prefill/dense/kernel/sm100_fmha_bwd_kernel_tma_warpspecialized.hpp
--- a/csrc/sm100/prefill/dense/kernel/sm100_fmha_bwd_mla_kernel_tma_warpspecialized.hpp
+++ b/csrc/sm100/prefill/dense/kernel/sm100_fmha_bwd_mla_kernel_tma_warpspecialized.hpp
--- a/csrc/sm100/prefill/dense/kernel/sm100_fmha_fwd_kernel_tma_warpspecialized.hpp
+++ b/csrc/sm100/prefill/dense/kernel/sm100_fmha_fwd_kernel_tma_warpspecialized.hpp
--- a/csrc/sm100/prefill/sparse/common_subroutine.h
+++ b/csrc/sm100/prefill/sparse/common_subroutine.h
--- a/csrc/sm100/prefill/sparse/fwd/head128/config.h
+++ b/csrc/sm100/prefill/sparse/fwd/head128/config.h
--- a/csrc/sm100/prefill/sparse/fwd/head128/instantiations/phase1_k512.cu
+++ b/csrc/sm100/prefill/sparse/fwd/head128/instantiations/phase1_k512.cu
-#include "../phase1.h"
-#include "../phase1.cuh"
-
-namespace sm100::fwd::head128 {
-
-template void run_fwd_phase1_kernel<512>(const SparseAttnFwdParams& params);
-
-}
--- a/csrc/sm100/prefill/sparse/fwd/head128/instantiations/phase1_k576.cu
+++ b/csrc/sm100/prefill/sparse/fwd/head128/instantiations/phase1_k576.cu
--- a/csrc/sm100/prefill/sparse/fwd/head128/phase1.cuh
+++ b/csrc/sm100/prefill/sparse/fwd/head128/phase1.cuh
--- a/csrc/sm100/prefill/sparse/fwd/head128/phase1.h
+++ b/csrc/sm100/prefill/sparse/fwd/head128/phase1.h