Check enablement of managed memory in hmm test and example code (#250)

ROCm · May 5, 2021 · 313bfc2 · 313bfc2
1 parent ffbc871
commit 313bfc2
Show file tree

Hide file tree

Showing 2 changed files with 23 additions and 0 deletions.
diff --git a/clients/include/testing_managed_malloc.hpp b/clients/include/testing_managed_malloc.hpp
@@ -181,6 +181,16 @@ void testing_managed_malloc(Arguments& argus)
 
     rocblas_int hot_calls = argus.iters;
 
+    // check managed memory enablement
+    int deviceID, hmm_enabled;
+    hipGetDevice(&deviceID);
+    hipDeviceGetAttribute(&hmm_enabled, hipDeviceAttributeManagedMemory, deviceID);
+    if(!hmm_enabled)
+    {
+        rocsolver_cout << "Managed memory not enabled on device. Skipping test..." << std::endl;
+        return;
+    }
+
     // check non-supported values
     // N/A
 

diff --git a/clients/samples/example_hmm.c b/clients/samples/example_hmm.c
@@ -56,6 +56,19 @@ int main() {
   rocblas_handle handle;
   rocblas_create_handle(&handle);
 
+  // check for enablement of managed memory
+  int deviceID, hmm_enabled;
+  hipGetDevice(&deviceID);
+  hipDeviceGetAttribute(&hmm_enabled,
+                        hipDeviceAttributeManagedMemory,
+                        deviceID);
+  if (!hmm_enabled)
+  {
+    printf("Managed memory not enabled on device %i\n", deviceID);
+    rocblas_destroy_handle(handle);
+    return 0;
+  }
+
   // calculate the sizes of our arrays
   size_t size_piv = (M < N) ? M : N; // count of Householder scalars