From e1de612405ea074cf913eef851dd7df2cd6d8779 Mon Sep 17 00:00:00 2001 From: Joseph Schuchart Date: Fri, 12 Apr 2024 01:13:20 -0400 Subject: [PATCH] Fix logic for detecting device-aware MPI Signed-off-by: Joseph Schuchart --- ttg/ttg/parsec/ttg.h | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/ttg/ttg/parsec/ttg.h b/ttg/ttg/parsec/ttg.h index 991e29664..3069922b5 100644 --- a/ttg/ttg/parsec/ttg.h +++ b/ttg/ttg/parsec/ttg.h @@ -3390,11 +3390,11 @@ ttg::abort(); // should not happen if (!need_pushout) { bool device_supported = false; if constexpr (derived_has_cuda_op()) { - device_supported = !world.impl().mpi_support(ttg::ExecutionSpace::CUDA); + device_supported = world.impl().mpi_support(ttg::ExecutionSpace::CUDA); } else if constexpr (derived_has_hip_op()) { - device_supported = !world.impl().mpi_support(ttg::ExecutionSpace::HIP); + device_supported = world.impl().mpi_support(ttg::ExecutionSpace::HIP); } else if constexpr (derived_has_level_zero_op()) { - device_supported = !world.impl().mpi_support(ttg::ExecutionSpace::L0); + device_supported = world.impl().mpi_support(ttg::ExecutionSpace::L0); } /* if MPI supports the device we don't care whether we have remote peers * because we can send from the device directly */