Use CUDA min/max built-in functions.

lattice-land · Aug 26, 2024 · b19c84e · b19c84e
1 parent eb46ab9
commit b19c84e
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/include/battery/utility.hpp b/include/battery/utility.hpp
@@ -113,23 +113,25 @@ CUDA inline int strcmp(const char* s1, const char* s2) {
   #endif
 }
 
-template<class T> CUDA constexpr T min(T a, T b) {
+template<class T> CUDA INLINE constexpr T min(T a, T b) {
   #ifdef __CUDA_ARCH__
     // When C++23 is available
     // if !consteval { return ::min(a, b); }
     // else { return std::min(a, b); }
-    return a < b ? a : b;
+    // return  a < b ? a : b;
+    return ::min(a, b);
   #else
     return std::min(a, b);
   #endif
 }
 
-template<class T> CUDA constexpr T max(T a, T b) {
+template<class T> CUDA INLINE constexpr T max(T a, T b) {
   #ifdef __CUDA_ARCH__
     // When C++23 is available
     // if !consteval { return ::max(a, b); }
     // else { return std::max(a, b); }
-    return a > b ? a : b;
+    // return a > b ? a : b;
+    return ::max(a, b); // a > b ? a : b;
   #else
     return std::max(a, b);
   #endif