-
Notifications
You must be signed in to change notification settings - Fork 10.7k
/
CMakeLists.txt
43 lines (36 loc) · 1.08 KB
/
CMakeLists.txt
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
31
32
33
34
35
36
37
38
39
40
41
42
43
macro(add_se_library name)
add_llvm_library(${name} ${ARGN})
set_target_properties(${name} PROPERTIES FOLDER "streamexecutor libraries")
endmacro(add_se_library)
if(STREAM_EXECUTOR_ENABLE_CUDA_PLATFORM)
set(
CMAKE_MODULE_PATH
${CMAKE_MODULE_PATH}
"${CMAKE_CURRENT_SOURCE_DIR}/platforms/cuda/cmake/modules/")
find_package(Libcuda REQUIRED)
include_directories(${LIBCUDA_INCLUDE_DIRS})
set(
STREAM_EXECUTOR_CUDA_PLATFORM_TARGET_OBJECT
$<TARGET_OBJECTS:streamexecutor_cuda_platform>)
set(
STREAM_EXECUTOR_LIBCUDA_LIBRARIES
${LIBCUDA_LIBRARIES})
endif(STREAM_EXECUTOR_ENABLE_CUDA_PLATFORM)
add_subdirectory(platforms)
add_se_library(
streamexecutor
Device.cpp
DeviceMemory.cpp
Error.cpp
HostMemory.cpp
Kernel.cpp
KernelSpec.cpp
PackedKernelArgumentArray.cpp
Platform.cpp
PlatformDevice.cpp
PlatformManager.cpp
Stream.cpp
${STREAM_EXECUTOR_CUDA_PLATFORM_TARGET_OBJECT}
LINK_LIBS
${STREAM_EXECUTOR_LIBCUDA_LIBRARIES})
install(TARGETS streamexecutor DESTINATION lib)