device_matrix_view_from_host patch for Grace

rapidsai · Dec 10, 2024 · cbe786a · cbe786a
1 parent e1a5708
commit cbe786a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/cpp/src/neighbors/detail/cagra/utils.hpp b/cpp/src/neighbors/detail/cagra/utils.hpp
@@ -184,7 +184,7 @@ class device_matrix_view_from_host {
     cudaPointerAttributes attr;
     RAFT_CUDA_TRY(cudaPointerGetAttributes(&attr, host_view.data_handle()));
     device_ptr = reinterpret_cast<T*>(attr.devicePointer);
-    if (device_ptr == NULL) {
+    if (attr.type != cudaMemoryTypeDevice && attr.type != cudaMemoryTypeManaged) {
       // allocate memory and copy over
       // NB: We use the temporary "large" workspace resource here; this structure is supposed to
       // live on stack and not returned to a user.