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

add ORT_DIRECTORY compiler option

Browse Source
This commit is contained in:
zhoushunjie
2022-09-21 15:01:47 +08:00
parent a8e447f225
commit 662b1e6113
7 changed files with 59 additions and 30 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
python/fastdeploy/c_lib_wrap.py.in
View File

@@ -16,7 +16,7 @@ import logging
import os
import sys
user_specified_dirs = ['@OPENCV_DIRECTORY@', ]
user_specified_dirs = ['@OPENCV_DIRECTORY@', '@ORT_DIRECTORY@', ]
def is_built_with_gpu() -> bool:

2
python/scripts/process_libraries.py.in
View File

@@ -18,7 +18,7 @@ import shutil
import subprocess
import platform
user_specified_dirs = ['@OPENCV_DIRECTORY@', ]
user_specified_dirs = ['@OPENCV_DIRECTORY@', '@ORT_DIRECTORY@', ]
def process_on_linux(current_dir):
rpaths = ["$ORIGIN:$ORIGIN/libs"]

1
python/setup.py
View File

@@ -63,6 +63,7 @@ setup_configs["CUDA_DIRECTORY"] = os.getenv("CUDA_DIRECTORY",
setup_configs["LIBRARY_NAME"] = PACKAGE_NAME
setup_configs["PY_LIBRARY_NAME"] = PACKAGE_NAME + "_main"
setup_configs["OPENCV_DIRECTORY"] = os.getenv("OPENCV_DIRECTORY", "")
setup_configs["ORT_DIRECTORY"] = os.getenv("ORT_DIRECTORY", "")
if setup_configs["WITH_GPU"] == "ON":
wheel_name = "fastdeploy-gpu-python"