Fix device comm code

Signed-off-by: Joseph Schuchart <schuchart@icl.utk.edu>
devreal · Sep 25, 2023 · 6d73621 · 6d73621
1 parent a7fe69a
commit 6d73621
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/ttg/ttg/parsec/ttg.h b/ttg/ttg/parsec/ttg.h
@@ -259,15 +259,15 @@ namespace ttg_parsec {
           || MPIX_Query_cuda_support()
 #endif // MPIX_CUDA_AWARE_SUPPORT
          ) {
-        mpi_space_support[static_cast<ttg::ExecutionSpace::CUDA>] = true;
+        mpi_space_support[static_cast<std::size_t>(ttg::ExecutionSpace::CUDA)] = true;
       }
 
       if (ttg::detail::force_device_comm()
 #if defined(MPIX_HIP_AWARE_SUPPORT) && MPIX_HIP_AWARE_SUPPORT
           || MPIX_Query_hip_support()
 #endif // MPIX_HIP_AWARE_SUPPORT
          ) {
-        mpi_space_support[static_cast<ttg::ExecutionSpace::HIP>] = true;
+        mpi_space_support[static_cast<std::size_t>(ttg::ExecutionSpace::HIP)] = true;
       }
 
 #if defined(PARSEC_PROF_TRACE)

diff --git 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