未验证 提交 c3d1522c 编写于 作者: Z zhupengyang 提交者: GitHub

fix python whl of armlinux (#3546)

上级 bdec7a4e
...@@ -33,11 +33,17 @@ else: ...@@ -33,11 +33,17 @@ else:
PADDLELITE_VERSION = PADDLELITE_TAG PADDLELITE_VERSION = PADDLELITE_TAG
# core lib of paddlelite is stored as lite.so # core lib of paddlelite is stored as lite.so
LITE_PATH = '${PADDLE_BINARY_DIR}/inference_lite_lib/python/install/lite' files = os.listdir('${PADDLE_BINARY_DIR}')
INFERENCE_LITE_LIB_PATH = ''
for file in files:
if file.find('inference_lite_lib') == 0:
INFERENCE_LITE_LIB_PATH = '${PADDLE_BINARY_DIR}/' + file
break
LITE_PATH = INFERENCE_LITE_LIB_PATH + '/python/install/lite'
PACKAGE_DATA = {'paddlelite': ['lite.so' if os.name!='nt' else 'lite.pyd']} PACKAGE_DATA = {'paddlelite': ['lite.so' if os.name!='nt' else 'lite.pyd']}
# put all thirdparty libraries in paddlelite.libs # put all thirdparty libraries in paddlelite.libs
PACKAGE_DATA['paddlelite.libs'] = [] PACKAGE_DATA['paddlelite.libs'] = []
LIB_PATH = '${PADDLE_BINARY_DIR}/inference_lite_lib/python/install/libs' LIB_PATH = INFERENCE_LITE_LIB_PATH + '/python/install/libs/'
if '${WITH_MKL}' == 'ON': if '${WITH_MKL}' == 'ON':
shutil.copy('${MKLML_SHARED_IOMP_LIB}', LIB_PATH) shutil.copy('${MKLML_SHARED_IOMP_LIB}', LIB_PATH)
shutil.copy('${MKLML_SHARED_LIB}', LIB_PATH) shutil.copy('${MKLML_SHARED_LIB}', LIB_PATH)
...@@ -49,8 +55,7 @@ if '${WITH_MKL}' == 'ON': ...@@ -49,8 +55,7 @@ if '${WITH_MKL}' == 'ON':
PACKAGE_DATA['paddlelite.libs'] += ['msvcr120.dll'] PACKAGE_DATA['paddlelite.libs'] += ['msvcr120.dll']
# link lite.so to paddlelite.libs # link lite.so to paddlelite.libs
if os.name != 'nt': if os.name != 'nt':
COMMAND = "patchelf --set-rpath '$ORIGIN/../libs/' ${PADDLE_BINARY_DIR}\ COMMAND = "patchelf --set-rpath '$ORIGIN/../libs/' " + LITE_PATH + "/lite.so"
/inference_lite_lib/python/install/lite/lite.so"
if os.system(COMMAND) != 0: if os.system(COMMAND) != 0:
raise Exception("patch third_party libs failed, command: %s" % COMMAND) raise Exception("patch third_party libs failed, command: %s" % COMMAND)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册