Set try_compile() target type

rapidsai · Nov 8, 2024 · a87fc27 · a87fc27
1 parent 1b80099
commit a87fc27
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/cpp/CMakeLists.txt b/cpp/CMakeLists.txt
@@ -85,6 +85,8 @@ else()
   set(cuFile_FOUND 1)
 
   # Check batch and stream API support (cuFile_BATCH_API_FOUND and cuFile_STREAM_API_FOUND)
+  set(_old_type ${CMAKE_TRY_COMPILE_TARGET_TYPE})
+  set(CMAKE_TRY_COMPILE_TARGET_TYPE STATIC_LIBRARY)
   try_compile(
     cuFile_BATCH_API_FOUND SOURCE_FROM_CONTENT
     batch.cpp
@@ -113,6 +115,7 @@ else()
   )
   message(STATUS "stream output: ${stream_output}")
   message(STATUS "Found cuFile Stream API: ${cuFile_STREAM_API_FOUND}")
+  set(CMAKE_TRY_COMPILE_TARGET_TYPE ${_old_type})
 endif()
 
 include(cmake/thirdparty/get_thread_pool.cmake)