blob: a422789a5ebaedf9c0c524b856e44d39ba5d5f8a (
plain)
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
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
|
SET(CMAKE_MODULE_PATH ${CMAKE_CURRENT_SOURCE_DIR}/../../cmake ${CMAKE_MODULE_PATH})
IF(NOT Torch_FOUND)
FIND_PACKAGE(Torch)
ENDIF()
IF(NOT CUDA_FOUND)
FIND_PACKAGE(CUDA 5.5 REQUIRED)
ENDIF()
IF(NOT MAGMA_FOUND)
FIND_PACKAGE(MAGMA)
ENDIF()
IF(MAGMA_FOUND)
SET(USE_MAGMA 1)
MESSAGE(STATUS "Compiling with MAGMA support")
ENDIF(MAGMA_FOUND)
IF(APPLE)
IF(${CUDA_VERSION} LESS 6.0)
# work around for mac os x bug:
# http://stackoverflow.com/questions/16286588/cuda-5-0-cmake-and-make-failing-on-osx-10-8-3
if (NOT DEFINED CUDA_HOST_COMPILER AND CMAKE_C_COMPILER_ID STREQUAL "Clang" AND EXISTS /usr/bin/gcc)
set(CUDA_HOST_COMPILER /usr/bin/gcc CACHE FILEPATH "Host side compiler used by NVCC")
message(STATUS "Setting CMAKE_HOST_COMPILER to /usr/bin/gcc instead of ${CMAKE_C_COMPILER}.")
endif()
# bug on Apple
LINK_DIRECTORIES("/usr/local/cuda/lib/")
ELSEIF(${CUDA_VERSION} LESS 7.0)
SET(CUDA_HOST_COMPILER clang)
LIST(APPEND CUDA_NVCC_FLAGS "-Xcompiler -stdlib=libstdc++ -Xlinker -stdlib=libstdc++")
IF("${CMAKE_CXX_COMPILER_ID}" STREQUAL "Clang")
set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -stdlib=libstdc++")
ENDIF()
ENDIF()
# CUDA 7 supports clang and libc++ so no need to change anything
ENDIF(APPLE)
# Detect CUDA architecture and get best NVCC flags
INCLUDE(${CMAKE_CURRENT_SOURCE_DIR}/../../cmake/FindCudaArch.cmake)
SELECT_NVCC_ARCH_FLAGS(NVCC_FLAGS_EXTRA)
LIST(APPEND CUDA_NVCC_FLAGS ${NVCC_FLAGS_EXTRA})
INCLUDE_DIRECTORIES(${CUDA_INCLUDE_DIRS})
INCLUDE_DIRECTORIES("${CUDA_SDK_ROOT_DIR}/common/inc")
IF(USE_MAGMA)
INCLUDE_DIRECTORIES(${MAGMA_INCLUDE_DIR})
ENDIF(USE_MAGMA)
IF(NOT THC_INSTALL_BIN_SUBDIR
OR NOT THC_INSTALL_LIB_SUBDIR
OR NOT THC_INSTALL_INCLUDE_SUBDIR
OR NOT THC_INSTALL_CMAKE_SUBDIR)
INCLUDE_DIRECTORIES(${TH_INCLUDE_PATH} ${TH_INCLUDE_PATH}/TH)
LINK_DIRECTORIES(${TH_LIB_PATH})
SET(THC_INSTALL_BIN_SUBDIR "bin" CACHE PATH "THC install binary subdirectory")
SET(THC_INSTALL_LIB_SUBDIR "lib" CACHE PATH "THC install library subdirectory")
SET(THC_INSTALL_INCLUDE_SUBDIR "include" CACHE PATH "THC install include subdirectory")
SET(THC_INSTALL_CMAKE_SUBDIR "share/cmake/THC" CACHE PATH "THC install cmake subdirectory")
ELSE()
SET(THC_INSTALL_BIN_SUBDIR ${Torch_INSTALL_BIN_SUBDIR})
SET(THC_INSTALL_LIB_SUBDIR ${Torch_INSTALL_LIB_SUBDIR})
SET(THC_INSTALL_INCLUDE_SUBDIR ${Torch_INSTALL_INCLUDE_SUBDIR})
SET(THC_INSTALL_CMAKE_SUBDIR ${Torch_INSTALL_CMAKE_SUBDIR})
ENDIF()
INCLUDE_DIRECTORIES("${CMAKE_CURRENT_BINARY_DIR}")
CONFIGURE_FILE(THCGeneral.h.in "${CMAKE_CURRENT_BINARY_DIR}/THCGeneral.h")
SET(CMAKE_C_FLAGS "-std=c99")
SET(src
THCGeneral.c THCAllocator.c THCStorage.c THCStorageCopy.c THCTensor.c THCTensorCopy.c)
SET(src-cuda
THCReduceApplyUtils.cu
THCBlas.cu
THCStorage.cu
THCStorageCopy.cu
THCTensor.cu
THCTensorCopy.cu
THCTensorMath.cu
THCTensorMath2.cu
THCTensorMathBlas.cu
THCTensorMathCompare.cu
THCTensorMathCompareT.cu
THCTensorMathMagma.cu
THCTensorMathPairwise.cu
THCTensorMathPointwise.cu
THCTensorMathScan.cu
THCTensorMathTransformReduce.cu
THCTensorMasked.cu
THCTensorIndex.cu
THCTensorConv.cu
THCTensorRandom.cu
THCTensorScatterGather.cu
THCApply.cu
THCTensorSort.cu
)
CUDA_ADD_LIBRARY(THC SHARED ${src} ${src-cuda})
CUDA_ADD_CUBLAS_TO_TARGET(THC)
TARGET_LINK_LIBRARIES(THC TH ${CUDA_curand_LIBRARY})
IF(USE_MAGMA)
TARGET_LINK_LIBRARIES(THC ${MAGMA_LIBRARIES})
ENDIF(USE_MAGMA)
INSTALL(TARGETS THC
RUNTIME DESTINATION "${THC_INSTALL_BIN_SUBDIR}"
LIBRARY DESTINATION "${THC_INSTALL_LIB_SUBDIR}"
ARCHIVE DESTINATION "${THC_INSTALL_LIB_SUBDIR}")
INSTALL(FILES
THC.h
${CMAKE_CURRENT_BINARY_DIR}/THCGeneral.h
THCBlas.h
THCStorage.h
THCStorageCopy.h
THCTensor.h
THCTensorCopy.h
THCTensorRandom.h
THCTensorMath.h
THCTensorConv.h
THCTensorSort.h
THCApply.cuh
THCReduce.cuh
THCReduceAll.cuh
THCReduceApplyUtils.cuh
THCAllocator.h
THCDeviceUtils.cuh
THCDeviceTensor.cuh
THCDeviceTensor-inl.cuh
THCDeviceTensorUtils.cuh
THCDeviceTensorUtils-inl.cuh
DESTINATION "${THC_INSTALL_INCLUDE_SUBDIR}/THC")
|
