From b450ed7e9cddf459546e385154b5f4d38b3ce7a2 Mon Sep 17 00:00:00 2001
From: Hashim Sharif <hsharif3@miranda.cs.illinois.edu>
Date: Sat, 27 Mar 2021 19:17:13 -0500
Subject: [PATCH] Adding check in set_path.sh throwing error if CUDA path not
 set

---
 hpvm/set_paths.sh | 29 ++++++++++++++++-------------
 1 file changed, 16 insertions(+), 13 deletions(-)

diff --git a/hpvm/set_paths.sh b/hpvm/set_paths.sh
index 42d1be5294..8c80a2a30b 100644
--- a/hpvm/set_paths.sh
+++ b/hpvm/set_paths.sh
@@ -1,17 +1,20 @@
 #!/bin/bash
 
 # These paths can be modified by the HPVM user
-CUDA_TOOLKIT_PATH=/software/cuda-9.1/
-CUDA_INCLUDE_PATH=$CUDA_TOOLKIT_PATH/include
-CUDA_LIB_PATH=$CUDA_TOOLKIT_PATH/lib64/
+CUDA_TOOLKIT_PATH=   # Set this to the root of your CUDA Installation
+if [ -z "$CUDA_TOOLKIT_PATH" ]
+then
+    echo "ERROR: SET CUDA_TOOLKIT_PATH to the Root of your CUDA Installation"
+else
+    CUDA_INCLUDE_PATH=$CUDA_TOOLKIT_PATH/include
+    CUDA_LIB_PATH=$CUDA_TOOLKIT_PATH/lib64/
+    echo "Setting environment paths..."
 
-echo "Setting environment paths..."
-
-# Setting CUDA paths here
-export CUDA_BIN_PATH=$CUDA_TOOLKIT_PATH
-export CUDA_INCLUDE_PATH=$CUDA_INCLUDE_PATH
-export CUDNN_PATH=$CUDA_LIB_PATH
-export LIBRARY_PATH=$CUDA_LIB_PATH:$LIBRARY_PATH
-export LD_LIBRARY_PATH=$CUDA_LIB_PATH:$LD_LIBRARY_PATH
-
-echo "Finished setting environment paths!"
+    # Setting CUDA paths here
+    export CUDA_BIN_PATH=$CUDA_TOOLKIT_PATH
+    export CUDA_INCLUDE_PATH=$CUDA_INCLUDE_PATH
+    export CUDNN_PATH=$CUDA_LIB_PATH
+    export LIBRARY_PATH=$CUDA_LIB_PATH:$LIBRARY_PATH
+    export LD_LIBRARY_PATH=$CUDA_LIB_PATH:$LD_LIBRARY_PATH
+    echo "Finished setting environment paths!"
+fi
-- 
GitLab