apps/FastDeploy
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/FastDeploy.git synced 2025-10-16 13:41:30 +08:00
Code Issues Actions 6 Packages Projects Releases Wiki Activity

Support Ascend deployment on X86 platform

Browse Source
This commit is contained in:
yunyaoXYY
2023-01-05 05:47:26 +00:00
parent eb0c8750f0
commit 26e98fc849
2 changed files with 51 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
scripts/patch_paddle_lite.py
View File

@@ -23,12 +23,15 @@ import sys
def process_paddle_lite(paddle_lite_so_path):
if platform.system().lower() != "linux":
return
rpaths = ["$ORIGIN"]
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)
@@ -37,7 +40,25 @@ def process_paddle_lite(paddle_lite_so_path):
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])