Update Bridges2 modules (#274)

MFlowCode · Dec 28, 2023 · 37ddb15 · 37ddb15
1 parent cf90cc0
commit 37ddb15
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 2 deletions.
diff --git a/mfc.sh b/mfc.sh
@@ -142,7 +142,14 @@ if [ "$1" == 'load' ]; then
         fi
     done
 
-    ok "All modules have been loaded for $M$COMPUTER$CR on $M$CG$CR"'s.'
+    if [ "$cg" == 'gpu' ]; then
+        isnv=$($FC --version | grep NVIDIA | wc -l)
+        if [ $isnv -eq 0 ]; then
+            export CC=nvc CXX=nvc++ FC=nvfortran
+        fi
+    fi
+    ok 'All modules and environment variables have been loaded.'
+
 
     return
 elif [ "$1" == "format" ]; then

diff --git a/toolchain/modules b/toolchain/modules
@@ -15,7 +15,7 @@ s-gpu nvhpc/22.11 cuda/nvhpc
 b     PSC Bridges2
 b-all python/3.8.6
 b-cpu allocations/1.0 gcc/10.2.0 openmpi/4.0.5-gcc10.2.0
-b-gpu openmpi/4.0.5-nvhpc22.9 nvhpc/22.9
+b-gpu openmpi/4.0.5-nvhpc22.9 nvhpc/22.9 cuda
 
 a     OLCF Ascent
 a-all python cmake/3.22.2