link THPP to THCS when CUDA available

soumith · soumith · commit 8ced72ccb840 · 2017-03-05T11:32:43.000-05:00
diff --git a/torch/lib/THPP/CMakeLists.txt b/torch/lib/THPP/CMakeLists.txt
@@ -52,6 +52,11 @@ ELSE()
       SET(THC_LIBRARIES "THC")
     ENDIF(NOT THC_LIBRARIES)
     MESSAGE(STATUS "THC_LIBRARIES: ${THC_LIBRARIES}")
+
+    IF(NOT THCS_LIBRARIES)
+      SET(THCS_LIBRARIES "THCS")
+    ENDIF(NOT THCS_LIBRARIES)
+    MESSAGE(STATUS "THCS_LIBRARIES: ${THCS_LIBRARIES}")
   ENDIF()
 ENDIF()
 
@@ -87,7 +92,7 @@ endif(NOT ${CMAKE_VERSION} VERSION_LESS "3.1")
 
 TARGET_LINK_LIBRARIES(THPP LINK_PRIVATE ${TH_LIBRARIES} ${THS_LIBRARIES})
 IF(CUDA_FOUND)
-  TARGET_LINK_LIBRARIES(THPP LINK_PRIVATE ${THC_LIBRARIES})
+  TARGET_LINK_LIBRARIES(THPP LINK_PRIVATE ${THC_LIBRARIES} ${THCS_LIBRARIES})
 ENDIF()
 
 INSTALL(TARGETS THPP
