From 83fc3c10cf0624145ff5eb735827e746a7d1af05 Mon Sep 17 00:00:00 2001
From: Michael Vines <mvines@gmail.com>
Date: Wed, 14 Nov 2018 19:19:27 -0800
Subject: [PATCH] Setup CUDA env for local builds

---
 net/net.sh                  | 5 +++++
 net/remote/remote-client.sh | 4 ++++
 net/remote/remote-sanity.sh | 5 +++++
 3 files changed, 14 insertions(+)

diff --git a/net/net.sh b/net/net.sh
index 6e5fd17b9b..8a61838b45 100755
--- a/net/net.sh
+++ b/net/net.sh
@@ -131,6 +131,11 @@ build() {
 
     set -x
     rm -rf farf
+
+    if [[ -r target/perf-libs/env.sh ]]; then
+      # shellcheck source=/dev/null
+      source target/perf-libs/env.sh
+    fi
     $MAYBE_DOCKER cargo install --features="$cargoFeatures" --root farf
     ./scripts/install-native-programs.sh farf/
   )
diff --git a/net/remote/remote-client.sh b/net/remote/remote-client.sh
index 5ad440cb5b..228d5f450c 100755
--- a/net/remote/remote-client.sh
+++ b/net/remote/remote-client.sh
@@ -39,6 +39,10 @@ local|tar)
   export USE_INSTALL=1
   export SOLANA_DEFAULT_METRICS_RATE=1
 
+  ./fetch-perf-libs.sh
+  # shellcheck source=/dev/null
+  source ./target/perf-libs/env.sh
+
   net/scripts/rsync-retry.sh -vPrc "$entrypointIp:~/.cargo/bin/solana*" ~/.cargo/bin/
   solana_bench_tps=solana-bench-tps
   solana_keygen=solana-keygen
diff --git a/net/remote/remote-sanity.sh b/net/remote/remote-sanity.sh
index 0e8c7dbe0c..f9f210e141 100755
--- a/net/remote/remote-sanity.sh
+++ b/net/remote/remote-sanity.sh
@@ -69,6 +69,11 @@ snap)
 local|tar)
   PATH="$HOME"/.cargo/bin:"$PATH"
   export USE_INSTALL=1
+  if [[ -r target/perf-libs/env.sh ]]; then
+    # shellcheck source=/dev/null
+    source target/perf-libs/env.sh
+  fi
+
   entrypointRsyncUrl="$entrypointIp:~/solana"
 
   solana_bench_tps=solana-bench-tps