From a6b74d57c5131d35fb0c97a5a57de38df15efb55 Mon Sep 17 00:00:00 2001
From: Graham Markall <gmarkall@nvidia.com>
Date: Mon, 4 Mar 2024 21:56:40 +0000
Subject: [PATCH] Build test binaries for current GPU

---
 ci/test_conda.sh | 2 ++
 ci/test_wheel.sh | 1 +
 2 files changed, 3 insertions(+)

diff --git a/ci/test_conda.sh b/ci/test_conda.sh
index 5751cfd..dddf861 100755
--- a/ci/test_conda.sh
+++ b/ci/test_conda.sh
@@ -36,6 +36,8 @@ rapids-mamba-retry install \
 
 rapids-logger "Build Tests"
 pushd test_binary_generation
+export GPU_CC=`nvidia-smi --query-gpu=compute_cap --format=csv | grep -v compute_cap | sed 's/\.//'
+`
 make
 popd
 
diff --git a/ci/test_wheel.sh b/ci/test_wheel.sh
index cf55a40..0b00e82 100755
--- a/ci/test_wheel.sh
+++ b/ci/test_wheel.sh
@@ -26,6 +26,7 @@ python -m pip install --find-links ./dist pynvjitlink${PACKAGE_CUDA_SUFFIX}
 
 rapids-logger "Build Tests"
 pushd test_binary_generation
+export GPU_CC=`nvidia-smi --query-gpu=compute_cap --format=csv | grep -v compute_cap | sed 's/\.//'
 make
 popd