rename

vectorch-ai · Feb 4, 2025 · 41dff63 · 41dff63
1 parent 60a6136
commit 41dff63
Show file tree

Hide file tree

Showing 8 changed files with 12 additions and 12 deletions.
diff --git a/src/kernels/attention/CMakeLists.txt b/src/kernels/attention/CMakeLists.txt
@@ -16,7 +16,7 @@ cc_library(
     mha_tile.h
     mha_traits_sm80.h
     mha_kernel_sm80.cuh
-    mha_launch_sm80.cuh
+    mha_dispatch_sm80.cuh
   DEPS
     cutlass
 )

diff --git a/src/kernels/attention/attn_api.cpp b/src/kernels/attention/attn_api.cpp
@@ -3,9 +3,9 @@
 #include <ATen/cuda/CUDAContext.h>
 
 #include "cute/layout.hpp"
+#include "mha_dispatch_sm80.cuh"
 #include "mha_params.h"
 #include "static_dispatch.h"
-#include "mha_launch_sm80.cuh"
 
 namespace llm {
 using namespace cute;

diff --git a/src/kernels/attention/mha_launch_sm80.cuh → src/kernels/attention/mha_dispatch_sm80.cuh b/src/kernels/attention/mha_launch_sm80.cuh → src/kernels/attention/mha_dispatch_sm80.cuh
diff --git a/src/kernels/attention/mha_kernel_sm80_pagedkv_test.cu b/src/kernels/attention/mha_kernel_sm80_pagedkv_test.cu
@@ -3,8 +3,8 @@
 #include <torch/torch.h>
 
 #include "cute/layout.hpp"
-#include "mha_kernel_sm80.cuh"
-#include "mha_launch_sm80.cuh"
+#include "mha_dispatch_sm80.cuh"
+#include "mha_kernel_sm80.cuh"  // IWYU pragma: keep
 #include "mha_params.h"
 #include "mha_ref.h"
 

diff --git a/src/kernels/attention/mha_kernel_sm80_test.cu b/src/kernels/attention/mha_kernel_sm80_test.cu
@@ -4,8 +4,8 @@
 #include <cstdint>
 
 #include "cute/layout.hpp"
-#include "mha_kernel_sm80.cuh"
-#include "mha_launch_sm80.cuh"
+#include "mha_dispatch_sm80.cuh"
+#include "mha_kernel_sm80.cuh"  // IWYU pragma: keep
 #include "mha_params.h"
 #include "mha_ref.h"
 

diff --git a/src/kernels/attention/mha_kernel_sm80_varlen_test.cu b/src/kernels/attention/mha_kernel_sm80_varlen_test.cu
@@ -3,8 +3,8 @@
 #include <torch/torch.h>
 
 #include "cute/layout.hpp"
-#include "mha_kernel_sm80.cuh"
-#include "mha_launch_sm80.cuh"
+#include "mha_dispatch_sm80.cuh"
+#include "mha_kernel_sm80.cuh"  // IWYU pragma: keep
 #include "mha_params.h"
 #include "mha_ref.h"
 

diff --git a/src/kernels/attention/mha_sm80_bench.cu b/src/kernels/attention/mha_sm80_bench.cu
@@ -4,8 +4,8 @@
 #include <cuda/std/chrono>
 #include <nvbench/nvbench.cuh>
 
-#include "mha_kernel_sm80.cuh"
-#include "mha_launch_sm80.cuh"
+#include "mha_dispatch_sm80.cuh"
+#include "mha_kernel_sm80.cuh"  // IWYU pragma: keep
 #include "mha_params.h"
 
 using namespace llm;

diff --git a/src/kernels/attention/mha_sm80_pagedkv_bench.cu b/src/kernels/attention/mha_sm80_pagedkv_bench.cu
@@ -5,8 +5,8 @@
 #include <cuda/std/chrono>
 #include <nvbench/nvbench.cuh>
 
-#include "mha_kernel_sm80.cuh"
-#include "mha_launch_sm80.cuh"
+#include "mha_dispatch_sm80.cuh"
+#include "mha_kernel_sm80.cuh"  // IWYU pragma: keep
 #include "mha_params.h"
 
 using namespace llm;