added udf for generic type in polynomials benchmark

rapidsai · Jan 8, 2025 · 0de2903 · 0de2903
1 parent 866979e
commit 0de2903
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/cpp/benchmarks/transform/polynomials.cpp b/cpp/benchmarks/transform/polynomials.cpp
@@ -64,10 +64,13 @@ static void BM_transform_polynomials(nvbench::state& state)
       expr = "( " + expr + " ) * x + " + std::to_string(constants[i + 1]);
     }
 
+    static_assert(std::is_same_v<key_type, float> || std::is_same_v<key_type, double>);
+    std::string type = std::is_same_v<key_type, float> ? "float" : "double";
+
     std::string udf = R"***(
 __device__ inline void    fdsf   (
-       float* out,
-       float x
+       )***" + type + R"***(* out,
+       )***" + type + R"***( x
 )
 {
   *out = )***" + expr +