From 75a276b3685da719efe1bee4909a30bf92fb4cf0 Mon Sep 17 00:00:00 2001
From: Olivier BICHLER <olivier.bichler@cea.fr>
Date: Tue, 26 Nov 2024 09:03:16 +0000
Subject: [PATCH] Added CUDA dependencies for aidge_quantization

---
 .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 19d311d..df934d8 100644
--- a/.gitlab/ci/default.gitlab-ci.yml
+++ b/.gitlab/ci/default.gitlab-ci.yml
@@ -51,6 +51,12 @@ variables:
           export LD_LIBRARY_PATH=/usr/local/cuda/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
           export CUDACXX=$(which nvcc)
           ;;
+        "aidge_quantization" )
+          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