File size: 778 Bytes
9dd3461 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 | if(TARGET caffe2::Threads)
return()
endif()
find_package(Threads REQUIRED)
# Threads::Threads doesn't work if the target has CUDA code
if(THREADS_FOUND)
add_library(caffe2::Threads INTERFACE IMPORTED)
if(THREADS_HAVE_PTHREAD_ARG)
set(compile_options
$<$<COMPILE_LANGUAGE:C>:-pthread>
$<$<COMPILE_LANGUAGE:CXX>:-pthread>)
if(USE_CUDA)
list(APPEND compile_options
$<$<COMPILE_LANGUAGE:CUDA>:-Xcompiler -pthread>)
endif()
set_property(TARGET caffe2::Threads
PROPERTY INTERFACE_COMPILE_OPTIONS
${compile_options})
endif()
if(CMAKE_THREAD_LIBS_INIT)
set_property(TARGET caffe2::Threads
PROPERTY INTERFACE_LINK_LIBRARIES "${CMAKE_THREAD_LIBS_INIT}")
endif()
endif()
|