From 57070eee3a5748e00039b212ed4897bafdf1a1e8 Mon Sep 17 00:00:00 2001
From: Nicolas Richart <nicolas.richart@epfl.ch>
Date: Mon, 9 Sep 2024 09:02:56 +0200
Subject: [PATCH] Adding forward compatibility for cuda
---
stacks/pinot-noir/modules_acc_cuda.yaml | 9 +++++++++
1 file changed, 9 insertions(+)
diff --git a/stacks/pinot-noir/modules_acc_cuda.yaml b/stacks/pinot-noir/modules_acc_cuda.yaml
index dd76dbb..d098e12 100644
--- a/stacks/pinot-noir/modules_acc_cuda.yaml
+++ b/stacks/pinot-noir/modules_acc_cuda.yaml
@@ -21,3 +21,12 @@ modules:
environment:
set:
NCCL_HOME: "{prefix}"
+ # needed as long as drivers are with 12.2 support
+ cuda@12.4:
+ environment:
+ prepend_path:
+ LD_LIBRARY_PATH: /ssoft/spack/external/cuda-compat/12.4/lib64
+ cuda@12.5:
+ environment:
+ prepend_path:
+ LD_LIBRARY_PATH: /ssoft/spack/external/cuda-compat/12.6/lib64
--
GitLab