Update Cmake infra to run HIP CXX tests using top-level cmake

Author: diptorupd

cmake/Options.cmake

@@ -112,22 +112,24 @@ if(FLASHINFER_ENABLE_HIP AND FLASHINFER_ENABLE_CUDA)
 endif()
 
 # Handle CUDA architectures
-if(FLASHINFER_CUDA_ARCHITECTURES)
-  message(STATUS "CMAKE_CUDA_ARCHITECTURES set to ${FLASHINFER_CUDA_ARCHITECTURES}.")
-else()
-  # No user-provided architectures, try to detect the CUDA archs based on where
-  # the project is being built
-  set(detected_archs "")
-  detect_cuda_architectures(detected_archs)
-  if(detected_archs)
-    set(FLASHINFER_CUDA_ARCHITECTURES ${detected_archs} CACHE STRING
-        "CUDA architectures" FORCE)
-    message(STATUS "Setting FLASHINFER_CUDA_ARCHITECTURES to detected values: ${FLASHINFER_CUDA_ARCHITECTURES}")
+if(FLASHINFER_ENABLE_CUDA)
+  if(FLASHINFER_CUDA_ARCHITECTURES)
+    message(STATUS "CMAKE_CUDA_ARCHITECTURES set to ${FLASHINFER_CUDA_ARCHITECTURES}.")
   else()
-    # No architectures detected, use safe defaults
-    set(FLASHINFER_CUDA_ARCHITECTURES "75;80;86" CACHE STRING
-        "CUDA architectures to compile for" FORCE)
-    message(STATUS "No architectures detected, using defaults: ${FLASHINFER_CUDA_ARCHITECTURES}")
+    # No user-provided architectures, try to detect the CUDA archs based on where
+    # the project is being built
+    set(detected_archs "")
+    detect_cuda_architectures(detected_archs)
+    if(detected_archs)
+      set(FLASHINFER_CUDA_ARCHITECTURES ${detected_archs} CACHE STRING
+          "CUDA architectures" FORCE)
+      message(STATUS "Setting FLASHINFER_CUDA_ARCHITECTURES to detected values: ${FLASHINFER_CUDA_ARCHITECTURES}")
+    else()
+      # No architectures detected, use safe defaults
+      set(FLASHINFER_CUDA_ARCHITECTURES "75;80;86" CACHE STRING
+          "CUDA architectures to compile for" FORCE)
+      message(STATUS "No architectures detected, using defaults: ${FLASHINFER_CUDA_ARCHITECTURES}")
+    endif()
   endif()
 endif()
 

cmake/utils/ConfigureTargets.cmake

@@ -1,7 +1,7 @@
 # cmake-format: off
 # Common configuration function for tests and benchmarks
 function(configure_flashinfer_target)
-  set(options IS_GTEST IS_BENCHMARK)
+  set(options IS_GTEST IS_BENCHMARK IS_HIP)
   set(oneValueArgs TARGET_NAME)
   set(multiValueArgs SOURCES LINK_LIBS COMPILE_FLAGS INCLUDE_DIRS)
 
@@ -27,6 +27,14 @@ function(configure_flashinfer_target)
   # Create executable target
   add_executable(${arg_TARGET_NAME} EXCLUDE_FROM_ALL ${arg_SOURCES})
 
+  if(arg_IS_HIP)
+    set_source_files_properties(${arg_SOURCES} PROPERTIES LANGUAGE HIP)
+    foreach(flag ${FLASHINFER_HIPCC_FLAGS})
+      target_compile_options(decode_kernels
+                             PRIVATE $<$<COMPILE_LANGUAGE:HIP>:${flag}>)
+    endforeach()
+  endif()
+
   # Add all include directories
   target_include_directories(
     ${arg_TARGET_NAME}