Merge pull request #2437 from typhoonzero/update_rdma_swig_cmake

Make swig link RDMA libs if RDMA enabled
8 years ago · 27b9389ae0
parent c4eb747f85 a85ca79a1d
commit 27b9389ae0
1 changed files with 2 additions and 0 deletions
--- a/paddle/api/CMakeLists.txt
+++ b/paddle/api/CMakeLists.txt
@ -41,6 +41,7 @@ SET(SWIG_MODULE_swig_paddle_EXTRA_DEPS
    paddle_network
    paddle_proto
    ${external_project_dependencies}
+    ${RDMA_LIBS}
 )

 IF(APPLE)
@ -73,6 +74,7 @@ SWIG_LINK_LIBRARIES(swig_paddle
    ${CMAKE_DL_LIBS}
    ${EXTERNAL_LIBS}
    ${CMAKE_THREAD_LIBS_INIT}
+    ${RDMA_LD_FLAGS}
    ${START_END}
 )