add cc 3.0 to nccl (#594)

pytorch · Jan 26, 2017 · a2938e3 · a2938e3
1 parent 2ad967d
commit a2938e3
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/torch/lib/nccl/Makefile b/torch/lib/nccl/Makefile
@@ -16,7 +16,8 @@ CUDA_LIB ?= $(CUDA_HOME)/lib64
 CUDA_INC ?= $(CUDA_HOME)/include
 NVCC ?= $(CUDA_HOME)/bin/nvcc
 
-NVCC_GENCODE ?= -gencode=arch=compute_35,code=sm_35 \
+NVCC_GENCODE ?= -gencode=arch=compute_30,code=sm_30 \
+                -gencode=arch=compute_35,code=sm_35 \
                 -gencode=arch=compute_50,code=sm_50 \
                 -gencode=arch=compute_52,code=sm_52 \
                 -gencode=arch=compute_52,code=compute_52