Hipified files should also get offload arch args

Author: danieldk

build2cmake/src/templates/cuda/kernel.cmake

@@ -53,7 +53,7 @@ elseif(GPU_LANG STREQUAL "HIP")
   message(STATUS "Archs for kernel {{kernel_name}}: {{ '${' + kernel_name + '_ARCHS}'}}")
 
   foreach(_KERNEL_SRC {{'${' + kernel_name + '_SRC}'}})
-    if(_KERNEL_SRC MATCHES ".*\\.hip$")
+    if(_KERNEL_SRC MATCHES ".*\\.(cu|hip)$")
       foreach(_ROCM_ARCH {{ '${' + kernel_name + '_ARCHS}'}})
         set_property(
           SOURCE ${_KERNEL_SRC}

build2cmake/src/templates/utils.cmake

@@ -74,6 +74,7 @@ function (hipify_sources_target OUT_SRCS NAME ORIG_SRCS)
   set(HIP_SRCS)
   foreach (SRC ${SRCS})
     get_source_file_property(include_dirs "${SRC}" INCLUDE_DIRECTORIES)
+    get_source_file_property(compile_options "${SRC}" COMPILE_OPTIONS)
     string(REGEX REPLACE "\.cu$" "\.hip" SRC ${SRC})
     string(REGEX REPLACE "cuda" "hip" SRC ${SRC})
 
@@ -84,6 +85,12 @@ function (hipify_sources_target OUT_SRCS NAME ORIG_SRCS)
         PROPERTIES INCLUDE_DIRECTORIES "${include_dirs}")
     endif()
 
+    if(compile_options)
+      set_source_files_properties(
+        ${SRC}
+        PROPERTIES COMPILE_OPTIONS "${compile_options}")
+    endif()
+
     list(APPEND HIP_SRCS "${CMAKE_CURRENT_BINARY_DIR}/${SRC}")
   endforeach()