Add SM90 kernels (#200)

2025-09-29 17:00:23 +08:00
parent 904b721731
commit 80ceeb2c76
10 changed files with 1602 additions and 0 deletions
--- a/csrc/indexing/main.cu
+++ b/csrc/indexing/main.cu
@@ -6,6 +6,11 @@
 #include <deep_gemm/impls/sm100_fp8_gemm_1d1d.cuh>
 #include <deep_gemm/impls/sm100_fp8_gemm_1d2d.cuh>

+// Attention kernels
+#include <deep_gemm/impls/sm90_fp8_mqa_logits.cuh>
+#include <deep_gemm/impls/sm90_fp8_paged_mqa_logits.cuh>
+#include <deep_gemm/impls/smxx_clean_logits.cuh>
+
 // Einsum kernels
 #include <deep_gemm/impls/sm90_bmk_bnk_mn.cuh>
 #include <deep_gemm/impls/sm100_bmk_bnk_mn.cuh>