We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 0f17fa6 commit 0c20d2eCopy full SHA for 0c20d2e
1 file changed
build.sh
@@ -195,8 +195,8 @@ NVCC="ccache $CUDA_HOME/bin/nvcc"
195
if [ -n "$NVCC_ARCH" ]; then
196
ARCH=$NVCC_ARCH
197
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}
+ GPU_CC=$(nvidia-smi --query-gpu=compute_cap --format=csv,noheader 2>/dev/null | head -1 | tr -d '.')
+ ARCH=${GPU_CC:+sm_$GPU_CC}
200
ARCH=${ARCH:-native}
201
else
202
ARCH=native
0 commit comments