Remove prelude

KernelTuner · Jul 22, 2024 · 986ca55 · 986ca55
1 parent 4278106
commit 986ca55
Show file tree

Hide file tree

Showing 9 changed files with 39 additions and 48 deletions.
diff --git a/docs/guides.rst b/docs/guides.rst
@@ -5,5 +5,4 @@ Guides
 
    guides/introduction.rst
    guides/promotion.rst
-   guides/prelude.rst
    guides/constant.rst
diff --git a/docs/guides/prelude.md b/docs/guides/prelude.md
diff --git a/examples/vector_add/main.cu b/examples/vector_add/main.cu
@@ -4,7 +4,7 @@
 #include <vector>
 
 #include "kernel_float.h"
-using namespace kernel_float::prelude;
+namespace kf = kernel_float;
 
 void cuda_check(cudaError_t code) {
     if (code != cudaSuccess) {

diff --git a/examples/vector_add_tiling/main.cu b/examples/vector_add_tiling/main.cu
@@ -5,7 +5,7 @@
 
 #include "kernel_float.h"
 #include "kernel_float/tiling.h"
-using namespace kernel_float::prelude;
+namespace kf = kernel_float;
 
 void cuda_check(cudaError_t code) {
     if (code != cudaSuccess) {

diff --git a/include/kernel_float.h b/include/kernel_float.h
@@ -4,13 +4,13 @@
 #include "kernel_float/base.h"
 #include "kernel_float/bf16.h"
 #include "kernel_float/binops.h"
+#include "kernel_float/constant.h"
 #include "kernel_float/conversion.h"
 #include "kernel_float/fp16.h"
 #include "kernel_float/iterate.h"
 #include "kernel_float/macros.h"
 #include "kernel_float/memory.h"
 #include "kernel_float/meta.h"
-#include "kernel_float/prelude.h"
 #include "kernel_float/reduce.h"
 #include "kernel_float/triops.h"
 #include "kernel_float/unops.h"

diff --git a/include/kernel_float/bf16.h b/include/kernel_float/bf16.h
@@ -223,6 +223,7 @@ KERNEL_FLOAT_BF16_CAST(
 #endif
 
 using bfloat16 = __nv_bfloat16;
+KERNEL_FLOAT_VECTOR_ALIAS(bfloat16x, __nv_bfloat16)
 //KERNEL_FLOAT_TYPE_ALIAS(float16x, __nv_bfloat16)
 //KERNEL_FLOAT_TYPE_ALIAS(f16x, __nv_bfloat16)
 

diff --git a/include/kernel_float/fp16.h b/include/kernel_float/fp16.h
@@ -171,6 +171,7 @@ KERNEL_FLOAT_FP16_CAST(unsigned long, __ull2half_rn(input), (unsigned long)(__ha
 KERNEL_FLOAT_FP16_CAST(unsigned long long, __ull2half_rn(input), __half2ull_rz(input));
 
 using half = __half;
+KERNEL_FLOAT_VECTOR_ALIAS(half, __half)
 //KERNEL_FLOAT_TYPE_ALIAS(float16x, __half)
 //KERNEL_FLOAT_TYPE_ALIAS(f16x, __half)
 

diff --git a/include/kernel_float/vector.h b/include/kernel_float/vector.h
@@ -323,6 +323,21 @@ template<typename T> using vec7 = vec<T, 7>;
 template<typename T> using vec8 = vec<T, 8>;
 // clang-format on
 
+#define KERNEL_FLOAT_VECTOR_ALIAS(NAME, T) \
+    template<size_t N>                     \
+    using NAME##1 = vec<T, 1>;             \
+    using NAME##2 = vec<T, 2>;             \
+    using NAME##3 = vec<T, 3>;             \
+    using NAME##4 = vec<T, 4>;             \
+    using NAME##5 = vec<T, 5>;             \
+    using NAME##6 = vec<T, 6>;             \
+    using NAME##7 = vec<T, 7>;             \
+    using NAME##8 = vec<T, 8>;
+
+KERNEL_FLOAT_VECTOR_ALIAS(int, int)
+KERNEL_FLOAT_VECTOR_ALIAS(float, float)
+KERNEL_FLOAT_VECTOR_ALIAS(double, double)
+
 /**
  * Create a vector from a variable number of input values.
  *

diff --git a/single_include/kernel_float.h b/single_include/kernel_float.h
@@ -16,8 +16,8 @@
 
 //================================================================================
 // this file has been auto-generated, do not modify its contents!
-// date: 2024-07-22 11:19:46.100972
-// git hash: 3b349be35a239a302f4b3fd9a366f74e48976cf5
+// date: 2024-07-22 11:31:53.132636
+// git hash: 4278106f2a14629445668d7e3684dbc8faf8b94d
 //================================================================================
 
 #ifndef KERNEL_FLOAT_MACROS_H
@@ -3853,6 +3853,21 @@ template<typename T> using vec7 = vec<T, 7>;
 template<typename T> using vec8 = vec<T, 8>;
 // clang-format on
 
+#define KERNEL_FLOAT_VECTOR_ALIAS(NAME, T) \
+    template<size_t N>                     \
+    using NAME##1 = vec<T, 1>;             \
+    using NAME##2 = vec<T, 2>;             \
+    using NAME##3 = vec<T, 3>;             \
+    using NAME##4 = vec<T, 4>;             \
+    using NAME##5 = vec<T, 5>;             \
+    using NAME##6 = vec<T, 6>;             \
+    using NAME##7 = vec<T, 7>;             \
+    using NAME##8 = vec<T, 8>;
+
+KERNEL_FLOAT_VECTOR_ALIAS(int, int)
+KERNEL_FLOAT_VECTOR_ALIAS(float, float)
+KERNEL_FLOAT_VECTOR_ALIAS(double, double)
+
 /**
  * Create a vector from a variable number of input values.
  *
@@ -4060,6 +4075,7 @@ KERNEL_FLOAT_FP16_CAST(unsigned long, __ull2half_rn(input), (unsigned long)(__ha
 KERNEL_FLOAT_FP16_CAST(unsigned long long, __ull2half_rn(input), __half2ull_rz(input));
 
 using half = __half;
+KERNEL_FLOAT_VECTOR_ALIAS(half, __half)
 //KERNEL_FLOAT_TYPE_ALIAS(float16x, __half)
 //KERNEL_FLOAT_TYPE_ALIAS(f16x, __half)
 
@@ -4293,6 +4309,7 @@ KERNEL_FLOAT_BF16_CAST(
 #endif
 
 using bfloat16 = __nv_bfloat16;
+KERNEL_FLOAT_VECTOR_ALIAS(bfloat16x, __nv_bfloat16)
 //KERNEL_FLOAT_TYPE_ALIAS(float16x, __nv_bfloat16)
 //KERNEL_FLOAT_TYPE_ALIAS(f16x, __nv_bfloat16)