make cuda installation customizable

Author: gregrodgers

bin/aomp_common_vars

@@ -18,6 +18,11 @@ AOMP_INSTALL_DIR=${AOMP}_${AOMP_VERSION_STRING}
 # Clang repo is $HOME/aomp/clang
 AOMP_REPOS=${AOMP_REPOS:-$HOME/git/aomp}
 
+CUDA=${CUDA:-/usr/local/cuda}
+CUDAT=${CUDAT:-$CUDA/targets}
+CUDAINCLUDE=${CUDAINCLUDE:-$CUDA/include}
+CUDABIN=${CUDAH:-$CUDA/bin}
+
 # Set list of default nvptx subarchitectures to build
 # Only Cuda 9 and above supports sm_70
 NVPTXGPUS_DEFAULT="30,35,50,60,61"

bin/build_openmp.sh

@@ -43,9 +43,9 @@ if [ "$PROC" == "aarch64" ] ; then
    NVPTXGPUS_DEFAULT=""
    NVPTXGPUS=""
 else
-   CUDAH=`find /usr/local/cuda/targets -type f -name "cuda.h" 2>/dev/null`
+   CUDAH=`find $CUDAT -type f -name "cuda.h" 2>/dev/null`
    if [ "$CUDAH" == "" ] ; then
-      CUDAH=`find /usr/local/cuda/include -type f -name "cuda.h" 2>/dev/null`
+      CUDAH=`find $CUDAINCLUDE -type f -name "cuda.h" 2>/dev/null`
    fi
    if [ "$CUDAH" == "" ] ; then
       echo
@@ -82,8 +82,8 @@ if [ ! -z `which "getconf"` ]; then
 fi
 
 GCCMIN=5
-if [ -f /usr/local/cuda/bin/nvcc ] ; then
-   CUDAVER=`/usr/local/cuda/bin/nvcc --version | grep compilation | cut -d" " -f5 | cut -d"." -f1 `
+if [ -f $CUDABIN/nvcc ] ; then
+   CUDAVER=`$CUDABIN/nvcc --version | grep compilation | cut -d" " -f5 | cut -d"." -f1 `
    echo "CUDA VERSION IS $CUDAVER"
    if [ $CUDAVER -gt 8 ] ; then
      GCCMIN=7