From 56873fbc71a7a644ec3acea1aea46efbe9361ad6 Mon Sep 17 00:00:00 2001
From: Olivier BICHLER <olivier.bichler@cea.fr>
Date: Wed, 19 Mar 2025 10:32:09 +0000
Subject: [PATCH] Added aidge_pruning
---
.gitlab/ci/default.gitlab-ci.yml | 6 ++++++
1 file changed, 6 insertions(+)
diff --git a/.gitlab/ci/default.gitlab-ci.yml b/.gitlab/ci/default.gitlab-ci.yml
index df934d8..f0ac720 100644
--- a/.gitlab/ci/default.gitlab-ci.yml
+++ b/.gitlab/ci/default.gitlab-ci.yml
@@ -57,6 +57,12 @@ variables:
export LD_LIBRARY_PATH=/usr/local/cuda/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
export CUDACXX=$(which nvcc)
;;
+ "aidge_pruning" )
+ apt-get install -y -qq libcudnn8-dev
+ export PATH=/usr/local/cuda/bin${PATH:+:${PATH}}
+ export LD_LIBRARY_PATH=/usr/local/cuda/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
+ export CUDACXX=$(which nvcc)
+ ;;
"aidge_backend_opencv" )
apt-get install -y -qq libopencv-dev
;;
--
GitLab