Adding functionality to check where a buffer is allocated for HIP mem…

…ory space
ornladios · May 4, 2023 · b9a811c · b9a811c
1 parent fe36dc2
commit b9a811c
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/source/adios2/helper/adiosKokkos.cpp b/source/adios2/helper/adiosKokkos.cpp
@@ -89,6 +89,15 @@ bool IsGPUbuffer(const void *ptr)
     {
         return true;
     }
+#endif
+#ifdef ADIOS2_HAVE_KOKKOS_HIP
+    hipError_t ret;
+    hipPointerAttribute_t attr;
+    ret = hipPointerGetAttributes(&attr, ptr);
+    if (ret == hipSuccess && attr.memoryType == hipMemoryTypeDevice)
+    {
+        return true;
+    }
 #endif
     return false;
 }