Use cuda::std::max instead of thrust::max

rapidsai · Jan 23, 2025 · 9383a09 · 9383a09
1 parent d02f199
commit 9383a09
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/cpp/src/traversal/extract_bfs_paths_impl.cuh b/cpp/src/traversal/extract_bfs_paths_impl.cuh
@@ -30,6 +30,7 @@
 
 #include <rmm/exec_policy.hpp>
 
+#include <cuda/std/functional>
 #include <thrust/binary_search.h>
 #include <thrust/count.h>
 #include <thrust/distance.h>
@@ -52,7 +53,7 @@ template <typename vertex_t>
 struct compute_max {
   vertex_t __device__ operator()(vertex_t lhs, vertex_t rhs)
   {
-    return thrust::max<vertex_t>(lhs, rhs);
+    return cuda::std::max<vertex_t>(lhs, rhs);
   }
 };
 

diff --git a/cpp/tests/utilities/check_utilities.hpp b/cpp/tests/utilities/check_utilities.hpp
@@ -97,7 +97,7 @@ struct device_nearly_equal {
   bool __device__ operator()(type_t lhs, type_t rhs) const
   {
     return std::abs(lhs - rhs) <
-           cuda::std::max(thrust::max(lhs, rhs) * threshold_ratio, threshold_magnitude);
+           cuda::std::max(cuda::std::max(lhs, rhs) * threshold_ratio, threshold_magnitude);
   }
 };