fix(nccl-tests): CUDA 13.0 supports only sm_90+ (Hopper/H100)
This commit is contained in:
@@ -96,11 +96,8 @@ SRC_DIR=$(ls -d nccl-tests-* 2>/dev/null | head -1)
|
|||||||
cd "$SRC_DIR"
|
cd "$SRC_DIR"
|
||||||
|
|
||||||
echo "=== building all_reduce_perf ==="
|
echo "=== building all_reduce_perf ==="
|
||||||
# CUDA 13.0 dropped support for compute_60 (Pascal); target Volta+ only.
|
# CUDA 13.0 supports only Hopper (sm_90) and newer.
|
||||||
GENCODE="-gencode=arch=compute_70,code=sm_70 \
|
GENCODE="-gencode=arch=compute_90,code=sm_90"
|
||||||
-gencode=arch=compute_80,code=sm_80 \
|
|
||||||
-gencode=arch=compute_86,code=sm_86 \
|
|
||||||
-gencode=arch=compute_90,code=sm_90"
|
|
||||||
make MPI=0 \
|
make MPI=0 \
|
||||||
NVCC="$NVCC" \
|
NVCC="$NVCC" \
|
||||||
CUDA_HOME="$CUDA_HOME" \
|
CUDA_HOME="$CUDA_HOME" \
|
||||||
|
|||||||
Reference in New Issue
Block a user