blob: f966dacab4d62ee47b8d9cb06ac32119e533539b (
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
|
# -*- cmake -*-
MACRO(ADD_TORCH_LIBRARY package type src)
IF ("${type}" STREQUAL "STATIC")
if ("${src}" MATCHES "cu$" OR "${src}" MATCHES "cu;")
CUDA_ADD_LIBRARY(${package} STATIC ${src})
else()
ADD_LIBRARY(${package} STATIC ${src})
endif()
ELSE()
if ("${src}" MATCHES "cu$" OR "${src}" MATCHES "cu;")
CUDA_ADD_LIBRARY(${package} ${type} ${src})
else()
ADD_LIBRARY(${package} ${type} ${src})
endif()
ENDIF()
ENDMACRO()
MACRO(ADD_TORCH_PACKAGE package src luasrc)
INCLUDE_DIRECTORIES(${CMAKE_CURRENT_SOURCE_DIR})
INCLUDE_DIRECTORIES(${Torch_LUA_INCLUDE_DIR})
### C/C++ sources
# As per CMake doc, macro arguments are not variables, so simple test syntax not working
IF(NOT "${src}" STREQUAL "")
ADD_TORCH_LIBRARY(${package} MODULE "${src}")
### Torch packages supposes libraries prefix is "lib"
SET_TARGET_PROPERTIES(${package} PROPERTIES
PREFIX "lib"
IMPORT_PREFIX "lib"
INSTALL_NAME_DIR "@executable_path/${Torch_INSTALL_BIN2CPATH}")
IF(APPLE)
SET_TARGET_PROPERTIES(${package} PROPERTIES
LINK_FLAGS "-undefined dynamic_lookup")
ENDIF()
IF (BUILD_STATIC OR "$ENV{STATIC_TH}" STREQUAL "YES")
ADD_TORCH_LIBRARY(${package}_static STATIC "${src}")
SET_TARGET_PROPERTIES(${package}_static PROPERTIES
COMPILE_FLAGS "-fPIC")
SET_TARGET_PROPERTIES(${package}_static PROPERTIES
PREFIX "lib" IMPORT_PREFIX "lib" OUTPUT_NAME "${package}")
ENDIF()
INSTALL(TARGETS ${package}
RUNTIME DESTINATION ${Torch_INSTALL_LUA_CPATH_SUBDIR}
LIBRARY DESTINATION ${Torch_INSTALL_LUA_CPATH_SUBDIR})
ENDIF(NOT "${src}" STREQUAL "")
### lua sources
IF(NOT "${luasrc}" STREQUAL "")
INSTALL(FILES ${luasrc}
DESTINATION ${Torch_INSTALL_LUA_PATH_SUBDIR}/${package})
ENDIF(NOT "${luasrc}" STREQUAL "")
ENDMACRO(ADD_TORCH_PACKAGE)
|
