From 269c8a324e886306e0c9b31ea605ce59b772fa1e Mon Sep 17 00:00:00 2001 From: Weiqun Zhang Date: Sun, 2 Feb 2025 15:00:56 -0800 Subject: [PATCH] Apply suggestions from code review --- Src/Base/AMReX_ParallelDescriptor.cpp | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) diff --git a/Src/Base/AMReX_ParallelDescriptor.cpp b/Src/Base/AMReX_ParallelDescriptor.cpp index 83d55bb273..88e11c3b32 100644 --- a/Src/Base/AMReX_ParallelDescriptor.cpp +++ b/Src/Base/AMReX_ParallelDescriptor.cpp @@ -1524,13 +1524,19 @@ Initialize () #if defined(OMPI_HAVE_MPI_EXT_ROCM) && OMPI_HAVE_MPI_EXT_ROCM use_gpu_aware_mpi = (bool) MPIX_Query_rocm_support(); #elif defined(MPICH) && defined(MPIX_GPU_SUPPORT_HIP) - use_gpu_aware_mpi = (bool) MPIX_Query_hip_support(); + int is_supported = 0; + if (MPIX_GPU_query_support(MPIX_GPU_SUPPORT_HIP, &is_supported) == MPI_SUCCESS) { + use_gpu_aware_mpi = (bool) is_supported; + } #endif #elif defined(AMREX_USE_SYCL) #if defined(MPICH) && defined(MPIX_GPU_SUPPORT_ZE) - use_gpu_aware_mpi = (bool) MPIX_Query_ze_support(); + int is_supported = 0; + if (MPIX_GPU_query_support(MPIX_GPU_SUPPORT_ZE, &is_supported) == MPI_SUCCESS) { + use_gpu_aware_mpi = (bool) is_supported; + } #endif #endif