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

[Other] Optimize backend selection strategy (#574)

Browse Source 
* Optimize backend selection strategy

* remove debug code

* Update runtime.cc

* Update picodet.cc

* Update yolox.cc

* fix ci
This commit is contained in:
Jason
2022-11-13 20:15:20 +08:00
committed by GitHub
parent eff33d0e87
commit 51818ed825
8 changed files with 120 additions and 118 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
python/setup.py
View File

@@ -45,9 +45,15 @@ import multiprocessing
with open(os.path.join(TOP_DIR, "python", "requirements.txt")) as fin:
REQUIRED_PACKAGES = fin.read()
if os.getenv("BUILD_ON_CPU", "OFF") == "ON":
os.environ["ENABLE_PADDLE_BACKEND"] = "ON"
os.environ["ENABLE_ORT_BACKEND"] = "ON"
os.environ["ENABLE_OPENVINO_BACKEND"] = "ON"
os.environ["ENABLE_VISION"] = "ON"
os.environ["ENABLE_TEXT"] = "ON"
os.environ["WITH_GPU"] = "OFF"
setup_configs = dict()
setup_configs["ENABLE_PADDLE_FRONTEND"] = os.getenv("ENABLE_PADDLE_FRONTEND",
"ON")
setup_configs["ENABLE_RKNPU2_BACKEND"] = os.getenv("ENABLE_RKNPU2_BACKEND",
"OFF")
setup_configs["ENABLE_ORT_BACKEND"] = os.getenv("ENABLE_ORT_BACKEND", "OFF")