add shared warpctc lib in whl

79a336b7 · Luo Tao · d08550fd · 79a336b7
隐藏空白更改
内联并排

Showing with 3 addition and 1 deletion

python/setup.py.in python/setup.py.in +3 -1

未找到文件。
--- 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}',