From 0913e2e82f7ce5ac8458ca46efd6da67894cc1be Mon Sep 17 00:00:00 2001
From: Akash Kothari <akashk4@tyler.cs.illinois.edu>
Date: Fri, 22 Jan 2021 00:02:46 -0600
Subject: [PATCH] Add a shell script to set CUDA paths

---
 hpvm/set_paths.sh | 17 +++++++++++++++++
 1 file changed, 17 insertions(+)
 create mode 100644 hpvm/set_paths.sh

diff --git a/hpvm/set_paths.sh b/hpvm/set_paths.sh
new file mode 100644
index 0000000000..cbbada8e29
--- /dev/null
+++ b/hpvm/set_paths.sh
@@ -0,0 +1,17 @@
+#!/bin/bash
+
+# These paths can be modified by the HPVM user
+CUDA_TOOLKIT_PATH=cuda-toolkit/9.1
+CUDA_INCLUDE_PATH=/software/cuda-9.1/include
+CUDA_LIB_PATH=/software/cuda-9.1/lib64/
+
+echo "Setting environment paths..."
+
+# Setting CUDA paths here
+module load $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!"
-- 
GitLab