summarylogtreecommitdiffstats
diff options
context:
space:
mode:
-rw-r--r--.SRCINFO1
-rw-r--r--PKGBUILD5
2 files changed, 5 insertions, 1 deletions
diff --git a/.SRCINFO b/.SRCINFO
index b9316a2b416a..97072a3d9a95 100644
--- a/.SRCINFO
+++ b/.SRCINFO
@@ -7,6 +7,7 @@ pkgbase = python-onnxruntime
license = MIT
makedepends = git
makedepends = cmake
+ makedepends = ninja
makedepends = gcc-libs
makedepends = glibc
makedepends = cxxopts
diff --git a/PKGBUILD b/PKGBUILD
index 315967c1b04a..e52daaf6d4a8 100644
--- a/PKGBUILD
+++ b/PKGBUILD
@@ -22,6 +22,7 @@ license=('MIT')
makedepends=(
'git'
'cmake'
+ 'ninja'
'gcc-libs'
'glibc'
'cxxopts'
@@ -194,6 +195,7 @@ build() {
-Donnxruntime_CUDA_HOME=/opt/cuda
-Donnxruntime_CUDNN_HOME=/usr
-Donnxruntime_USE_NCCL=ON
+ -Donnxruntime_NVCC_THREADS=1
)
fi
@@ -206,7 +208,8 @@ build() {
cmake -S onnxruntime/cmake -B build \
"${_cmake_args[@]}" \
- "$@"
+ "$@" \
+ -G Ninja
LC_ALL=C cmake --build build #-v