Perlmutter Make: CUDA_PATH (#2059)

* Perlmutter: find CUDA_ROOT or CUDA_PATH. * Cover all bases.
AMReX-Codes · May 26, 2021 · db8e479 · db8e479
1 parent affbd9f
commit db8e479
Showing 1 changed file with 17 additions and 1 deletion.
diff --git a/Tools/GNUMake/sites/Make.nersc b/Tools/GNUMake/sites/Make.nersc
@@ -44,7 +44,23 @@ ifeq ($(which_computer),$(filter $(which_computer),perlmutter))
     endif
   endif
 
-  CUDA_ARCH = 80
+  ifeq ($(USE_CUDA),TRUE)
+    CUDA_ARCH = 80
+
+    ifneq ($(CUDA_ROOT),)
+        SYSTEM_CUDA_PATH := $(CUDA_ROOT)
+        COMPILE_CUDA_PATH := $(CUDA_ROOT)
+    else ifneq ($(CUDA_HOME),)
+        SYSTEM_CUDA_PATH := $(CUDA_HOME)
+        COMPILE_CUDA_PATH := $(CUDA_HOME)
+    else ifneq ($(CUDA_PATH),)
+        SYSTEM_CUDA_PATH := $(CUDA_PATH)
+        COMPILE_CUDA_PATH := $(CUDA_PATH)
+    else
+        $(error No CUDA_ROOT nor CUDA_HOME nor CUDA_PATH found. Please load a cuda module.)
+    endif
+
+  endif
 
 endif