Use cuda::std::byte in device functions

rapidsai · Oct 9, 2024 · b4e2b3f · b4e2b3f
1 parent 974bb1a
commit b4e2b3f
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/cpp/src/structure/remove_multi_edges_impl.cuh b/cpp/src/structure/remove_multi_edges_impl.cuh
@@ -28,6 +28,7 @@
 
 #include <rmm/device_uvector.hpp>
 
+#include <cuda/std/cstddef>
 #include <thrust/binary_search.h>
 #include <thrust/distance.h>
 #include <thrust/iterator/zip_iterator.h>
@@ -54,7 +55,7 @@ struct hash_src_dst_pair {
     pair[0] = thrust::get<0>(t);
     pair[1] = thrust::get<1>(t);
     cuco::murmurhash3_32<vertex_t*> hash_func{};
-    return hash_func.compute_hash(reinterpret_cast<std::byte*>(pair), 2 * sizeof(vertex_t)) %
+    return hash_func.compute_hash(reinterpret_cast<cuda::std::byte*>(pair), 2 * sizeof(vertex_t)) %
            num_groups;
   }
 };