diff options
| author | Soumith Chintala <soumith@gmail.com> | 2017-06-07 07:17:05 +0300 |
|---|---|---|
| committer | Soumith Chintala <soumith@gmail.com> | 2017-06-07 07:17:05 +0300 |
| commit | c2fa913b6a9bf91051f9b1f09393af1c7b13d7b5 (patch) | |
| tree | b72dccb3b5337e02b7357db1380f042fd6140564 | |
| parent | 8272995026e4d928961c6dfbd5c736f2143c224f (diff) | |
standalone macros
| -rw-r--r-- | lib/TH/CMakeLists.txt | 21 |
1 files changed, 21 insertions, 0 deletions
diff --git a/lib/TH/CMakeLists.txt b/lib/TH/CMakeLists.txt index b481a8a..bfc17cc 100644 --- a/lib/TH/CMakeLists.txt +++ b/lib/TH/CMakeLists.txt @@ -20,6 +20,27 @@ IF(NOT TH_INSTALL_BIN_SUBDIR SET(TH_INSTALL_CMAKE_SUBDIR "share/cmake/TH" CACHE PATH "TH install cmake subdirectory") ENDIF() +###################################################################### +###### macros section +##################################################################### +IF(NOT ADD_TORCH_LIBRARY) +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() +ENDIF() + ####################################################################### ##### flags section ###################################################################### |