add shared warpctc lib in whl

8 years ago · 79a336b78f
parent d08550fdd2
commit 79a336b78f
1 changed files with 3 additions and 1 deletions
--- a/python/setup.py.in
+++ b/python/setup.py.in
@ -31,7 +31,9 @@ paddle_bins = ['${PADDLE_BINARY_DIR}/paddle/scripts/paddle_usage',
               '${PADDLE_BINARY_DIR}/paddle/pserver/paddle_pserver_main']

 paddle_rt_lib_dir = 'local/lib'
-paddle_rt_libs = [] if '${MKL_SHARED_LIBS}'== '' else '${MKL_SHARED_LIBS}'.split(';')
+paddle_rt_libs = ['${WARPCTC_LIBRARIES}']
+if '${MKL_SHARED_LIBS}'!= '':
+  paddle_rt_libs += '${MKL_SHARED_LIBS}'.split(';')

 setup(name='paddlepaddle',
      version='${PADDLE_VERSION}',