diff --git a/projects/rccl/Makefile b/projects/rccl/Makefile index 79e85d32c9..49f28947c2 100644 --- a/projects/rccl/Makefile +++ b/projects/rccl/Makefile @@ -36,12 +36,12 @@ CUDA_LIB ?= $(CUDA_HOME)/lib64 CUDA_INC ?= $(CUDA_HOME)/include NVCC ?= $(CUDA_HOME)/bin/nvcc -CUDACODE ?= -gencode=arch=compute_35,code=sm_35 \ - -gencode=arch=compute_50,code=sm_50 \ - -gencode=arch=compute_52,code=sm_52 +NVCC_GENCODE ?= -gencode=arch=compute_35,code=sm_35 \ + -gencode=arch=compute_50,code=sm_50 \ + -gencode=arch=compute_52,code=sm_52 CXXFLAGS := -I$(CUDA_INC) -fPIC -fvisibility=hidden -NVCUFLAGS := -ccbin $(CXX) $(CUDACODE) -lineinfo -std=c++11 -maxrregcount 96 +NVCUFLAGS := -ccbin $(CXX) $(NVCC_GENCODE) -lineinfo -std=c++11 -maxrregcount 96 # Use addprefix so that we can specify more than one path LDFLAGS := $(addprefix -L,${CUDA_LIB}) -lcudart