From a6eae2e7f2bf03f268fcb6b055201ff6827684c4 Mon Sep 17 00:00:00 2001 From: Tim Dettmers Date: Wed, 20 Oct 2021 19:15:47 -0700 Subject: Added skip_zeros; tests are passing. --- Makefile | 48 +++++++++++++++++++++++++----------------------- 1 file changed, 25 insertions(+), 23 deletions(-) (limited to 'Makefile') diff --git a/Makefile b/Makefile index 1a5f17f..4fbe918 100644 --- a/Makefile +++ b/Makefile @@ -15,29 +15,31 @@ INCLUDE := -I $(CUDA_HOME)/include -I $(ROOT_DIR)/csrc -I $(CONDA_PREFIX)/inclu LIB := -L $(CUDA_HOME)/lib64 -lcudart -lcuda -lcublas -lcurand -lcusparse -L $(CONDA_PREFIX)/lib # NVIDIA NVCC compilation flags -COMPUTE_CAPABILITY := -gencode arch=compute_35,code=sm_35 # Kepler -COMPUTE_CAPABILITY += -gencode arch=compute_37,code=sm_37 # Kepler -COMPUTE_CAPABILITY += -gencode arch=compute_50,code=sm_50 # Maxwell -COMPUTE_CAPABILITY += -gencode arch=compute_52,code=sm_52 # Maxwell -COMPUTE_CAPABILITY += -gencode arch=compute_60,code=sm_60 # Pascal -COMPUTE_CAPABILITY += -gencode arch=compute_61,code=sm_61 # Pascal -COMPUTE_CAPABILITY += -gencode arch=compute_70,code=sm_70 # Volta -COMPUTE_CAPABILITY += -gencode arch=compute_72,code=sm_72 # Volta -COMPUTE_CAPABILITY += -gencode arch=compute_72,code=sm_72 # Volta - -# CUDA 9.2 supports CC 3.0, but CUDA >= 11.0 does not -CC_CUDA92 := -gencode arch=compute_30,code=sm_30 - -# Later versions of CUDA support the new architectures -CC_CUDA10x := -gencode arch=compute_30,code=sm_30 -CC_CUDA10x += -gencode arch=compute_75,code=sm_75 - -CC_CUDA110 := -gencode arch=compute_75,code=sm_75 -CC_CUDA110 += -gencode arch=compute_80,code=sm_80 - -CC_CUDA11x := -gencode arch=compute_75,code=sm_75 -CC_CUDA11x += -gencode arch=compute_80,code=sm_80 -CC_CUDA11x += -gencode arch=compute_86,code=sm_86 +#COMPUTE_CAPABILITY := -gencode arch=compute_35,code=sm_35 # Kepler +#COMPUTE_CAPABILITY += -gencode arch=compute_37,code=sm_37 # Kepler +#COMPUTE_CAPABILITY += -gencode arch=compute_50,code=sm_50 # Maxwell +#COMPUTE_CAPABILITY += -gencode arch=compute_52,code=sm_52 # Maxwell +#COMPUTE_CAPABILITY += -gencode arch=compute_60,code=sm_60 # Pascal +#COMPUTE_CAPABILITY += -gencode arch=compute_61,code=sm_61 # Pascal +#COMPUTE_CAPABILITY += -gencode arch=compute_70,code=sm_70 # Volta +#COMPUTE_CAPABILITY += -gencode arch=compute_72,code=sm_72 # Volta +#COMPUTE_CAPABILITY += -gencode arch=compute_72,code=sm_72 # Volta +# +## CUDA 9.2 supports CC 3.0, but CUDA >= 11.0 does not +#CC_CUDA92 := -gencode arch=compute_30,code=sm_30 +# +## Later versions of CUDA support the new architectures +#CC_CUDA10x := -gencode arch=compute_30,code=sm_30 +#CC_CUDA10x += -gencode arch=compute_75,code=sm_75 +# +#CC_CUDA110 := -gencode arch=compute_75,code=sm_75 +#CC_CUDA110 += -gencode arch=compute_80,code=sm_80 +# +#CC_CUDA11x := -gencode arch=compute_75,code=sm_75 +#CC_CUDA11x += -gencode arch=compute_80,code=sm_80 +#CC_CUDA11x += -gencode arch=compute_86,code=sm_86 + +COMPUTE_CAPABILITY := -gencode arch=compute_70,code=sm_70 # Volta all: $(ROOT_DIR)/dependencies/cub $(BUILD_DIR) -- cgit v1.2.3