From 7c4f3d325d65dbfab29185ced93ff096dd1bf374 Mon Sep 17 00:00:00 2001
From: Starbeamrainbowlabs <sbrl@starbeamrainbowlabs.com>
Date: Thu, 14 Nov 2024 22:28:39 +0000
Subject: [PATCH] slurm/dlr: fix workaround logic

---
 aimodel/slurm-TEST-deeplabv3p-rainfall.job | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/aimodel/slurm-TEST-deeplabv3p-rainfall.job b/aimodel/slurm-TEST-deeplabv3p-rainfall.job
index 7dafb48..103ce79 100755
--- a/aimodel/slurm-TEST-deeplabv3p-rainfall.job
+++ b/aimodel/slurm-TEST-deeplabv3p-rainfall.job
@@ -22,7 +22,7 @@ command_exists() {
 #######################################################
 
 # Fix "Could not load library libcublasLt.so.12. Error: libcublasLt.so.12: cannot open shared object file: No such file or directory" error
-if [[ "${SLURM_CLUSTER_NAME}" != "cs-cluster" ]] && [[ -d "${HOME}/cuda" ]]; then
+if [[ "${SLURM_CLUSTER_NAME}" == "cs-cluster" ]] && [[ -d "${HOME}/cuda" ]]; then
 	echo "[slurm_runner] csgpu cluster detected, sourcing extra CUDA setup script" >&2;
 	#shellcheck source=/dev/null
 	source "${HOME}/cuda/activate.sh";