summaryrefslogtreecommitdiff
path: root/Makefile
diff options
context:
space:
mode:
authorTim Dettmers <dettmers@cs.washington.edu>2021-10-20 19:15:47 -0700
committerTim Dettmers <dettmers@cs.washington.edu>2021-10-20 19:15:47 -0700
commita6eae2e7f2bf03f268fcb6b055201ff6827684c4 (patch)
treed2f72792251c9feaef1cf9dcddc3c79e6312a93a /Makefile
parentbb34fd50a1fec74e62beb6e23d51f0142c7d0ab6 (diff)
Added skip_zeros; tests are passing.
Diffstat (limited to 'Makefile')
-rw-r--r--Makefile48
1 files changed, 25 insertions, 23 deletions
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)