From 6d736213b7c284e247e06bff7455c56a43fd60cf Mon Sep 17 00:00:00 2001 From: Joseph Schuchart Date: Mon, 25 Sep 2023 20:52:44 +0000 Subject: [PATCH] Fix device comm code Signed-off-by: Joseph Schuchart --- ttg/ttg/parsec/ttg.h | 4 ++-- ttg/ttg/util/env.h | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/ttg/ttg/parsec/ttg.h b/ttg/ttg/parsec/ttg.h index 3bfad35a7..288545dbb 100644 --- a/ttg/ttg/parsec/ttg.h +++ b/ttg/ttg/parsec/ttg.h @@ -259,7 +259,7 @@ namespace ttg_parsec { || MPIX_Query_cuda_support() #endif // MPIX_CUDA_AWARE_SUPPORT ) { - mpi_space_support[static_cast] = true; + mpi_space_support[static_cast(ttg::ExecutionSpace::CUDA)] = true; } if (ttg::detail::force_device_comm() @@ -267,7 +267,7 @@ namespace ttg_parsec { || MPIX_Query_hip_support() #endif // MPIX_HIP_AWARE_SUPPORT ) { - mpi_space_support[static_cast] = true; + mpi_space_support[static_cast(ttg::ExecutionSpace::HIP)] = true; } #if defined(PARSEC_PROF_TRACE) diff --git a/ttg/ttg/util/env.h b/ttg/ttg/util/env.h index 2cbb20fea..4e07bff7b 100644 --- a/ttg/ttg/util/env.h +++ b/ttg/ttg/util/env.h @@ -22,7 +22,7 @@ namespace ttg { /// support this extension, users can force the use of device buffers in communication by setting /// `TTG_FORCE_DEVICE_COMM` to a non-negative number. /// @return true if the user wants to force the use of device-side buffers in communicaton. - bool enable_device_comm(); + bool force_device_comm(); } // namespace detail } // namespace ttg