From 771478bc688adf259c73fe015607a8776b450ecf Mon Sep 17 00:00:00 2001
From: Michael Vines <mvines@gmail.com>
Date: Thu, 21 Jun 2018 13:19:46 -0700
Subject: [PATCH] Add simple CUDA version check, warn on mismatch

---
 fetch-perf-libs.sh | 15 +++++++++++++++
 1 file changed, 15 insertions(+)

diff --git a/fetch-perf-libs.sh b/fetch-perf-libs.sh
index 0b92f50e8..00901665d 100755
--- a/fetch-perf-libs.sh
+++ b/fetch-perf-libs.sh
@@ -17,6 +17,21 @@ fi
   tar zxvf solana-perf.tgz
 )
 
+if [[ -r /usr/local/cuda/version.txt && -r cuda-version.txt ]]; then
+  if ! diff /usr/local/cuda/version.txt cuda-version.txt > /dev/null; then
+    echo ==============================================
+    echo Warning: possible CUDA version mismatch
+    echo
+    echo "Expected version: $(cat cuda-version.txt)"
+    echo "Detected version: $(cat /usr/local/cuda/version.txt)"
+    echo ==============================================
+  fi
+else
+  echo ==============================================
+  echo Warning: unable to validate CUDA version
+  echo ==============================================
+fi
+
 echo "Downloaded solana-perf version: $(cat solana-perf-HEAD.txt)"
 
 exit 0