From 9e7b9487b0668c784bf2fc8109bcc4072fe3c347 Mon Sep 17 00:00:00 2001
From: Michael Vines <mvines@gmail.com>
Date: Mon, 12 Nov 2018 17:50:16 -0800
Subject: [PATCH] perf-libs now drives setting CUDA_HOME

---
 ci/test-stable-perf.sh    |  5 ++---
 fetch-perf-libs.sh        | 18 +++++++++++++++++-
 net/remote/remote-node.sh |  6 ++----
 3 files changed, 21 insertions(+), 8 deletions(-)

diff --git a/ci/test-stable-perf.sh b/ci/test-stable-perf.sh
index c60bc8505..b5ab63b3e 100755
--- a/ci/test-stable-perf.sh
+++ b/ci/test-stable-perf.sh
@@ -14,11 +14,10 @@ if ! ci/version-check.sh stable; then
 fi
 export RUST_BACKTRACE=1
 export RUSTFLAGS="-D warnings"
-export CUDA_HOME=/usr/local/cuda
 
 ./fetch-perf-libs.sh
-export LD_LIBRARY_PATH=$PWD/target/perf-libs:$CUDA_HOME/lib64:$LD_LIBRARY_PATH
-export PATH=$PATH:$CUDA_HOME/bin
+# shellcheck source=/dev/null
+source ./target/perf-libs/env.sh
 
 _() {
   echo "--- $*"
diff --git a/fetch-perf-libs.sh b/fetch-perf-libs.sh
index 8d5abc20f..f8e9d1b50 100755
--- a/fetch-perf-libs.sh
+++ b/fetch-perf-libs.sh
@@ -19,7 +19,13 @@ mkdir -p target/perf-libs
     curl https://solana-perf.s3.amazonaws.com/v0.10.5/x86_64-unknown-linux-gnu/solana-perf.tgz | tar zxvf -
   )
 
-  : "${CUDA_HOME:=/usr/local/cuda}"
+  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
+  fi
 
   if [[ -r "$CUDA_HOME"/version.txt && -r cuda-version.txt ]]; then
     if ! diff "$CUDA_HOME"/version.txt cuda-version.txt > /dev/null; then
@@ -36,7 +42,17 @@ mkdir -p target/perf-libs
     echo ==============================================
   fi
 
+  cat > env.sh <<EOF
+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)"
+  echo
+  echo "source ./target/perf-libs/env.sh to setup compatible build environment"
 )
 
 exit 0
diff --git a/net/remote/remote-node.sh b/net/remote/remote-node.sh
index 72f6a614f..17e8b04ae 100755
--- a/net/remote/remote-node.sh
+++ b/net/remote/remote-node.sh
@@ -84,11 +84,9 @@ local|tar)
   export RUST_LOG
   export SOLANA_DEFAULT_METRICS_RATE=1
 
-  : "${CUDA_HOME:=/usr/local/cuda}"
-
   ./fetch-perf-libs.sh
-  export LD_LIBRARY_PATH="$PWD/target/perf-libs:$CUDA_HOME/lib64:$LD_LIBRARY_PATH"
-  echo "LD_LIBRARY_PATH=$LD_LIBRARY_PATH"
+  # shellcheck source=/dev/null
+  source ./target/perf-libs/env.sh
 
   scripts/oom-monitor.sh  > oom-monitor.log 2>&1 &
   scripts/net-stats.sh  > net-stats.log 2>&1 &