Skip to content

Commit 0c20d2e

Browse files
committed
build fix
1 parent 0f17fa6 commit 0c20d2e

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

build.sh

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -195,8 +195,8 @@ NVCC="ccache $CUDA_HOME/bin/nvcc"
195195
if [ -n "$NVCC_ARCH" ]; then
196196
ARCH=$NVCC_ARCH
197197
elif command -v nvidia-smi &>/dev/null; then
198-
CC=$(nvidia-smi --query-gpu=compute_cap --format=csv,noheader 2>/dev/null | head -1 | tr -d '.')
199-
ARCH=${CC:+sm_$CC}
198+
GPU_CC=$(nvidia-smi --query-gpu=compute_cap --format=csv,noheader 2>/dev/null | head -1 | tr -d '.')
199+
ARCH=${GPU_CC:+sm_$GPU_CC}
200200
ARCH=${ARCH:-native}
201201
else
202202
ARCH=native

0 commit comments

Comments
 (0)