From b2c1ca825acccafeac1ca7a1ed6a77460457b28a Mon Sep 17 00:00:00 2001
From: agsantos <aline.gondimsantos@savoirfairelinux.com>
Date: Mon, 4 Jan 2021 16:37:06 -0500
Subject: [PATCH] contrib: add onnxruntime
Change-Id: I4b8d8d6dde97b12ec64a853c640e173c187d6aa4
---
contrib/src/onnx/package.json | 17 +++++++++++++++++
contrib/src/onnx/rules.mak | 28 ++++++++++++++++++++++++++++
2 files changed, 45 insertions(+)
create mode 100644 contrib/src/onnx/package.json
create mode 100644 contrib/src/onnx/rules.mak
diff --git a/contrib/src/onnx/package.json b/contrib/src/onnx/package.json
new file mode 100644
index 0000000000..0f8a34b6b8
--- /dev/null
+++ b/contrib/src/onnx/package.json
@@ -0,0 +1,17 @@
+{
+ "name": "onnx",
+ "version": "v1.6.0",
+ "url": "https://github.com/microsoft/onnxruntime/archive/__VERSION__.tar.gz",
+ "deps": [],
+ "patches": [],
+ "win_patches": [],
+ "project_paths": [],
+ "with_env" : "",
+ "custom_scripts": {
+ "pre_build": [
+ "mkdir build"
+ ],
+ "build": [],
+ "post_build": []
+ }
+}
diff --git a/contrib/src/onnx/rules.mak b/contrib/src/onnx/rules.mak
new file mode 100644
index 0000000000..f5b8bc0afc
--- /dev/null
+++ b/contrib/src/onnx/rules.mak
@@ -0,0 +1,28 @@
+# ONNX
+ONNX_VERSION := v1.6.0
+
+onnx: toolchain.cmake
+
+.onnx: onnx
+ if test -d "./onnx"; then ( rm -rf ./onnx && tar -xf ./../tarballs/onnxruntime-$(ONNX_VERSION).tar.gz ) else ( git clone https://github.com/microsoft/onnxruntime.git onnx && cd ./onnx && git checkout $(ONNX_VERSION) && cd ./../ && tar cf onnxruntime-$(ONNX_VERSION).tar.gz onnx && mv onnxruntime-$(ONNX_VERSION).tar.gz ../tarballs/) fi
+ifdef HAVE_ANDROID
+ cd $< && sh build.sh --parallel --android --android_sdk_path $(ANDROID_SDK) --android_ndk_path $(ANDROID_NDK) --android_abi $(ANDROID_ABI) --android_api 29 --use_nnapi --config Release --build_shared_lib --skip_tests --android_cpp_shared --minimal_build extended
+ cd $< && cp ./build/Linux/Release/libonnxruntime.so $(PREFIX)/lib/
+else
+ifdef USE_NVIDIA
+ cd $< && sh ./build.sh --config Release --build_shared_lib --parallel --use_cuda --cuda_version $(CUDA_VERSION) --cuda_home $(CUDA_PATH) --cudnn_home $(CUDNN_PATH) --skip_tests
+ if [ ! -d "$(PREFIX)/lib/onnxruntime" ] ; then (mkdir $(PREFIX)/lib/onnxruntime) fi
+ if [ ! -d "$(PREFIX)/lib/onnxruntime/nvidia-gpu" ] ; then (mkdir $(PREFIX)/lib/onnxruntime/nvidia-gpu) fi
+ cd $< && cp ./build/Linux/Release/libonnxruntime.so $(PREFIX)/lib/onnxruntime/nvidia-gpu/libonnxruntime.so
+else
+ cd $< && sh ./build.sh --config Release --build_shared_lib --parallel --skip_tests
+ if [ ! -d "$(PREFIX)/lib/onnxruntime" ] ; then (mkdir $(PREFIX)/lib/onnxruntime) fi
+ if [ ! -d "$(PREFIX)/lib/onnxruntime/cpu" ] ; then (mkdir $(PREFIX)/lib/onnxruntime/cpu) fi
+ cd $< && cp ./build/Linux/Release/libonnxruntime.so $(PREFIX)/lib/onnxruntime/cpu/libonnxruntime.so
+endif
+endif
+ if [ ! -d "$(PREFIX)/include/onnxruntime" ] ; then ( mkdir $(PREFIX)/include/onnxruntime ) fi
+ cd $< && cp -r ./include/onnxruntime/core/* $(PREFIX)/include/onnxruntime/
+ touch $@
+
+
--
GitLab