cmake: add CMake language support for CUDA/HIP

kokkos · Apr 10, 2024 · 7392588 · 7392588
1 parent 4f3220c
commit 7392588
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/CMakeLists.txt b/CMakeLists.txt
@@ -127,6 +127,9 @@ ELSE()
   IF (NOT KOKKOSKERNELS_HAS_TRILINOS AND NOT KOKKOSKERNELS_HAS_PARENT)
     # This is a standalone build
     FIND_PACKAGE(Kokkos REQUIRED)
+    IF(Kokkos_COMPILE_LANGUAGE)
+      ENABLE_LANGUAGE(${Kokkos_COMPILE_LANGUAGE})
+    ENDIF()
     IF(${Kokkos_VERSION} VERSION_GREATER_EQUAL "4.2.00")
       MESSAGE(STATUS "Found Kokkos version ${Kokkos_VERSION} at ${Kokkos_DIR}")
       IF((${Kokkos_VERSION} VERSION_GREATER "4.3.99"))

diff --git a/cmake/kokkoskernels_tribits.cmake b/cmake/kokkoskernels_tribits.cmake
@@ -85,6 +85,11 @@ ENDIF()
 IF(PARSE_SOURCES)
   LIST(REMOVE_DUPLICATES PARSE_SOURCES)
 ENDIF()
+IF(Kokkos_COMPILE_LANGUAGE)
+  FOREACH(source ${PARSE_SOURCES})
+    SET_SOURCE_FILES_PROPERTIES(${source} PROPERTIES LANGUAGE ${Kokkos_COMPILE_LANGUAGE})
+  ENDFOREACH()
+ENDIF()
 
 ADD_LIBRARY(
   ${LIBRARY_NAME}