MCA/SSHMEM/UCX: Added warning messages for fallbacks

open-mpi · Aug 24, 2023 · 2fe0dcd · 2fe0dcd
1 parent 2858a93
commit 2fe0dcd
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/oshmem/mca/sshmem/ucx/sshmem_ucx_module.c b/oshmem/mca/sshmem/ucx/sshmem_ucx_module.c
@@ -189,8 +189,18 @@ segment_create(map_segment_t *ds_buf,
             ds_buf->allocator   = &sshmem_ucx_allocator;
             return OSHMEM_SUCCESS;
         }
+        else if(hint) {
+            SSHMEM_WARN("Hint was not recognized therefore ignored,"
+                        "fallback to host memory allocation");
+        } else {
+            SSHMEM_WARN("Failed to allocate DEVICE_NIC_MEM,"
+                        "fallback to host memory");
+        }
         /* Fallback - Try again using host memory*/
     }
+#else
+    SSHMEM_WARN("UCX version is too old and won't support"
+                "SHMEM_HINT_DEVICE_NIC_MEM hint, fallback to host memory");
 #endif
 
     flags |= UCP_MEM_MAP_FIXED;