Allow hip sources to be directly included when compiling for rocm. (v…

…llm-project#12087) Signed-off-by: ice-tong <[email protected]>
ice-tong · Jan 18, 2025 · a322a76 · a322a76
1 parent 79b8ba7
commit a322a76
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/cmake/utils.cmake b/cmake/utils.cmake
@@ -58,8 +58,8 @@ function (hipify_sources_target OUT_SRCS NAME ORIG_SRCS)
   #
   set(SRCS ${ORIG_SRCS})
   set(CXX_SRCS ${ORIG_SRCS})
-  list(FILTER SRCS EXCLUDE REGEX "\.(cc)|(cpp)$")
-  list(FILTER CXX_SRCS INCLUDE REGEX "\.(cc)|(cpp)$")
+  list(FILTER SRCS EXCLUDE REGEX "\.(cc)|(cpp)|(hip)$")
+  list(FILTER CXX_SRCS INCLUDE REGEX "\.(cc)|(cpp)|(hip)$")
 
   #
   # Generate ROCm/HIP source file names from CUDA file names.