kanshan/nccl-tests
1
0
Fork 0
mirror of https://github.com/NVIDIA/nccl-tests.git synced 2026-04-23 16:08:20 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Use Family-Specific architecture features on CUDA 13, enable Thor and Spark support

Browse Source
This commit is contained in:
jasl 2025-09-17 17:48:35 +08:00
parent 9a5c15461a
commit 0890dff640
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
src/common.mk
View File

@ -31,9 +31,9 @@ ifeq ($(shell test "0$(CUDA_MAJOR)" -ge 13; echo $$?),0)
NVCC_GENCODE ?= -gencode=arch=compute_75,code=sm_75 \
-gencode=arch=compute_80,code=sm_80 \
-gencode=arch=compute_90,code=sm_90 \
-gencode=arch=compute_100,code=sm_100 \
-gencode=arch=compute_120,code=sm_120 \
-gencode=arch=compute_120,code=compute_120
-gencode=arch=compute_100f,code=sm_100 \
-gencode=arch=compute_110f,code=sm_110 \
-gencode=arch=compute_120f,code=sm_120
else ifeq ($(shell test "0$(CUDA_MAJOR)" -eq 12 -a "0$(CUDA_MINOR)" -ge 8; echo $$?),0)
# Include Blackwell support if we're using CUDA12.8 or above
NVCC_GENCODE ?= -gencode=arch=compute_60,code=sm_60 \