diff --git a/Makefile b/Makefile index 79e85d3..49f2894 100644 --- a/Makefile +++ b/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