Merge pull request #3503 from luotao1/warpctc

add shared warpctc lib in whl
8 years ago · fdd7cad4d5
parent 0f8688192c 2d2cc2f5ce
commit fdd7cad4d5
1 changed files with 3 additions and 1 deletions
--- a/python/setup.py.in
+++ b/python/setup.py.in
@ -32,7 +32,9 @@ paddle_bins = ['${PADDLE_BINARY_DIR}/paddle/scripts/paddle_usage',
               '${PADDLE_BINARY_DIR}/paddle/scripts/paddle']

 paddle_rt_lib_dir = '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}',