Correct path to solana-perf-CUDA_HOME.txt

takeuchi5512 · Nov 13, 2018 · d0ebee5 · d0ebee5
1 parent aa7c741
commit d0ebee5
Showing 1 changed file with 5 additions and 8 deletions.
diff --git a/fetch-perf-libs.sh b/fetch-perf-libs.sh
@@ -19,14 +19,13 @@ mkdir -p target/perf-libs
     curl https://solana-perf.s3.amazonaws.com/v0.10.6/x86_64-unknown-linux-gnu/solana-perf.tgz | tar zxvf -
   )
 
-  if [[ -z "$CUDA_HOME" ]]; then
-    if [[ -r "$CUDA_HOME"/solana-perf-CUDA_HOME.txt ]]; then
-      CUDA_HOME=$(cat "$CUDA_HOME"/solana-perf-CUDA_HOME.txt)
-    else
-      CUDA_HOME=/usr/local/cuda
-    fi
+  if [[ -r solana-perf-CUDA_HOME.txt ]]; then
+    CUDA_HOME=$(cat solana-perf-CUDA_HOME.txt)
+  else
+    CUDA_HOME=/usr/local/cuda
   fi
 
+  echo CUDA_HOME="$CUDA_HOME"
   if [[ -r "$CUDA_HOME"/version.txt && -r cuda-version.txt ]]; then
     if ! diff "$CUDA_HOME"/version.txt cuda-version.txt > /dev/null; then
         echo ==============================================
@@ -46,8 +45,6 @@ mkdir -p target/perf-libs
 export CUDA_HOME=$CUDA_HOME
 export LD_LIBRARY_PATH="$PWD:$CUDA_HOME/lib64:$LD_LIBRARY_PATH"
 export PATH="$PATH:$CUDA_HOME/bin"
-echo CUDA_HOME="$CUDA_HOME"
-echo LD_LIBRARY_PATH="$LD_LIBRARY_PATH"
 EOF
 
   echo "Downloaded solana-perf version: $(cat solana-perf-HEAD.txt)"