apps/FastDeploy
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/FastDeploy.git synced 2025-10-06 00:57:33 +08:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

[Bug Fix]Fix KunlunXin compile bug (#1120)

Browse Source 
* fix KunlunXin compile bugs

* update code
This commit is contained in:
yeliang2258
2023-01-12 15:52:33 +08:00
committed by GitHub
parent dd1419ad7e
commit c866377382
1 changed files with 11 additions and 34 deletions
Download Patch File Download Diff File Expand all files Collapse all files

45
scripts/patch_paddle_lite.py Normal file → Executable file
View File

@@ -25,42 +25,19 @@ def process_paddle_lite(paddle_lite_so_path):
return
rpaths = ["$ORIGIN", "$ORIGIN/mklml/lib/"]
patchelf_exe = os.getenv("PATCHELF_EXE", "patchelf")
for paddle_lite_so_file in os.listdir(paddle_lite_so_path):
paddle_lite_so_file = os.path.join(paddle_lite_so_path,
paddle_lite_so_file)
# Patch /paddlelite/lib/*.so
if '.so' in paddle_lite_so_file:
command = "{} --set-rpath '{}' {}".format(
patchelf_exe, ":".join(rpaths), paddle_lite_so_file)
if platform.machine() != 'sw_64' and platform.machine(
) != 'mips64':
assert os.system(
command) == 0, "patchelf {} failed, the command: {}".format(
for root, dirs, files in os.walk(paddle_lite_so_path):
for lib in files:
if ".so" in lib:
paddle_lite_so_file = os.path.join(root, lib)
command = "{} --set-rpath '{}' {}".format(
patchelf_exe, ":".join(rpaths), paddle_lite_so_file)
if platform.machine() != 'sw_64' and platform.machine(
) != 'mips64':
assert os.system(
command
) == 0, "patchelf {} failed, the command: {}".format(
paddle_lite_so_file, command)
# Patch /paddlelite/lib/mklml/lib/*.so
if 'mklml' in paddle_lite_so_file:
paddle_lite_mklml_lib_path = os.path.join(
paddle_lite_so_path, paddle_lite_so_file, 'lib')
for paddle_lite_mklml_so_file in os.listdir(
paddle_lite_mklml_lib_path):
paddle_lite_mklml_so_file = os.path.join(
paddle_lite_mklml_lib_path, paddle_lite_mklml_so_file)
if '.so' in paddle_lite_mklml_so_file:
command = "{} --set-rpath '{}' {}".format(
patchelf_exe, ":".join(rpaths),
paddle_lite_mklml_so_file)
if platform.machine() != 'sw_64' and platform.machine(
) != 'mips64':
assert os.system(
command
) == 0, "patchelf {} failed, the command: {}".format(
paddle_lite_mklml_so_file, command)
if __name__ == "__main__":
process_paddle_lite(sys.argv[1])