Make: Switch to .cuh extension for CUDA

ashvardanian · Jan 16, 2025 · 37c0581 · 37c0581
1 parent f311a61
commit 37c0581
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/.vscode/settings.json b/.vscode/settings.json
@@ -96,13 +96,16 @@
     },
     "cSpell.words": [
         "ashvardanian",
+        "blas",
         "CCCL",
+        "constexpr",
         "CUDA",
         "Kahan",
         "openmp",
         "shfl",
         "SPIR",
         "STREQUAL",
-        "Vardanian"
+        "Vardanian",
+        "wmma"
     ]
 }
diff --git a/reduce_cublas.hpp → reduce_cublas.cuh b/reduce_cublas.hpp → reduce_cublas.cuh
@@ -58,8 +58,8 @@ struct cuda_blas_t {
     }
 };
 
-__global__ void cu_reduce_tensors(float const *inputs, unsigned int input_size, float *sums_per_row,
-                                  unsigned int columns) {
+__global__ void cu_reduce_tensors( //
+    float const *inputs, unsigned int input_size, float *sums_per_row, unsigned int columns) {
 
     // Tile using a 2D grid
     unsigned int first_row_of_block = blockIdx.x * blockDim.x;