diff --git a/src/common.mk b/src/common.mk index 5fd9418..d6e5c18 100644 --- a/src/common.mk +++ b/src/common.mk @@ -19,7 +19,14 @@ CUDA_MINOR = $(shell echo $(CUDA_VERSION) | cut -d "." -f 2) # Better define NVCC_GENCODE in your environment to the minimal set # of archs to reduce compile time. -ifeq ($(shell test "0$(CUDA_MAJOR)" -eq 12 -a "0$(CUDA_MINOR)" -ge 8 -o "0$(CUDA_MAJOR)" -ge 13; echo $$?),0) +ifeq ($(shell test "0$(CUDA_MAJOR)" -ge 13; echo $$?),0) +# Add Blackwell but drop Volta support if we're using CUDA13.0 or above +NVCC_GENCODE ?= -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 +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_70,code=sm_70 \ -gencode=arch=compute_80,code=sm_80 \