blob: 88457ba499a50f28b3299222c97313fc3068cb4f (
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
|
# -*- 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()
INSTALL(TARGETS ${package} DESTINATION ${LUALIBDIR})
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")
IF(APPLE)
SET_TARGET_PROPERTIES(${package} PROPERTIES
LINK_FLAGS "-undefined dynamic_lookup")
ENDIF()
SET_TARGET_PROPERTIES(${package} PROPERTIES
COMPILE_FLAGS "-fPIC")
SET_TARGET_PROPERTIES(${package} PROPERTIES
PREFIX "lib" IMPORT_PREFIX "lib" OUTPUT_NAME "${package}")
INSTALL(TARGETS ${package} DESTINATION ${LUALIBDIR})
ENDIF(NOT "${src}" STREQUAL "")
### lua sources
IF(NOT "${luasrc}" STREQUAL "")
INSTALL(FILES ${luasrc}
DESTINATION ${LUALIBDIR}/${package})
ENDIF(NOT "${luasrc}" STREQUAL "")
ENDMACRO(ADD_TORCH_PACKAGE)
|
