Build test binaries for current GPU

rapidsai · Mar 4, 2024 · a6b74d5 · a6b74d5
1 parent a3a32a7
commit a6b74d5
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git 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
@@ -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