Minor fixes to device cide

Signed-off-by: Joseph Schuchart <schuchart@icl.utk.edu>
devreal · Sep 29, 2023 · 3fa8001 · 3fa8001
1 parent 3aff11c
commit 3fa8001
Show file tree

Hide file tree

Showing 4 changed files with 7 additions and 2 deletions.
diff --git a/ttg/CMakeLists.txt b/ttg/CMakeLists.txt
@@ -130,6 +130,9 @@ if (TTG_HAVE_HIPBLAS)
     )
   endif (NOT TTG_HAVE_CUDA)
   list(APPEND ttg-deps roc::hipblas)
+  find_package(hip REQUIRED)
+  list(APPEND ttg-deps hip::host)
+
 endif (TTG_HAVE_HIPBLAS)
 
 set(ttg-public-headers ${ttg-headers};${ttg-impl-headers};${ttg-base-headers};${ttg-util-headers};${ttg_device_headers})

diff --git a/ttg/ttg/config.in.h b/ttg/ttg/config.in.h
@@ -15,6 +15,8 @@
 
 #cmakedefine TTG_HAVE_CUDART
 
+#cmakedefine TTG_HAVE_HIP
+
 #cmakedefine TTG_HAVE_HIPBLAS
 
 #cmakedefine TTG_HAVE_MPI

diff --git a/ttg/ttg/device/cublas_helper.cpp b/ttg/ttg/device/cublas_helper.cpp
@@ -10,7 +10,7 @@
 #endif // TTG_HAVE_CUDART
 
 #ifdef TTG_HAVE_HIPBLAS
-#include <hip.h>
+#include <hip_runtime.h>
 #endif // TTG_HAVE_HIPBLAS
 
 namespace ttg::detail {

diff --git a/ttg/ttg/device/task.h b/ttg/ttg/device/task.h
@@ -206,7 +206,7 @@ namespace ttg {
     // clang-format on
     template <size_t i, typename keyT, typename valueT>
     inline auto send(const keyT &key, valueT &&value) {
-      return send(i, key, std::forward<valueT>(value));
+      return ttg::device::send(i, key, std::forward<valueT>(value));
     }