try to fix OOM killer

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