File tree Expand file tree Collapse file tree 1 file changed +8
-1
lines changed Expand file tree Collapse file tree 1 file changed +8
-1
lines changed Original file line number Diff line number Diff line change @@ -19,7 +19,14 @@ CUDA_MINOR = $(shell echo $(CUDA_VERSION) | cut -d "." -f 2)
1919
2020# Better define NVCC_GENCODE in your environment to the minimal set
2121# of archs to reduce compile time.
22- ifeq ($(shell test "0$(CUDA_MAJOR ) " -eq 12 -a "0$(CUDA_MINOR ) " -ge 8 -o "0$(CUDA_MAJOR ) " -ge 13; echo $$? ) ,0)
22+ ifeq ($(shell test "0$(CUDA_MAJOR ) " -ge 13; echo $$? ) ,0)
23+ # Add Blackwell but drop Volta support if we're using CUDA13.0 or above
24+ NVCC_GENCODE ?= -gencode=arch=compute_80,code=sm_80 \
25+ -gencode=arch=compute_90,code=sm_90 \
26+ -gencode=arch=compute_100,code=sm_100 \
27+ -gencode=arch=compute_120,code=sm_120 \
28+ -gencode=arch=compute_120,code=compute_120
29+ else ifeq ($(shell test "0$(CUDA_MAJOR)" -eq 12 -a "0$(CUDA_MINOR)" -ge 8; echo $$?),0)
2330# Include Blackwell support if we're using CUDA12.8 or above
2431NVCC_GENCODE ?= -gencode=arch=compute_70,code=sm_70 \
2532 -gencode=arch=compute_80,code=sm_80 \
You can’t perform that action at this time.
0 commit comments