From 649f04d07711440ec2699ce7ec1af4ab71d6dbf9 Mon Sep 17 00:00:00 2001 From: Boris Fomitchev Date: Fri, 24 Mar 2017 11:58:14 -0700 Subject: [PATCH] Added Pascal nvcc flags, bumped version --- Makefile | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/Makefile b/Makefile index d0fe9b9..8f34fcb 100644 --- a/Makefile +++ b/Makefile @@ -20,7 +20,9 @@ NVCC ?= $(CUDA_HOME)/bin/nvcc NVCC_GENCODE ?= -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 + -gencode=arch=compute_60,code=sm_60\ + -gencode=arch=compute_61,code=sm_61 \ + -gencode=arch=compute_60,code=compute_60 CXXFLAGS := -I$(CUDA_INC) -fPIC -fvisibility=hidden NVCUFLAGS := -ccbin $(CXX) $(NVCC_GENCODE) -lineinfo -std=c++11 -maxrregcount 96 @@ -52,7 +54,7 @@ endif NCCL_MAJOR := 1 NCCL_MINOR := 3 -NCCL_PATCH := 3 +NCCL_PATCH := 4 CXXFLAGS += -DNCCL_MAJOR=$(NCCL_MAJOR) -DNCCL_MINOR=$(NCCL_MINOR) -DNCCL_PATCH=$(NCCL_PATCH) CUDA_VERSION ?= $(shell ls $(CUDA_LIB)/libcudart.so.* | head -1 | rev | cut -d "." -f -2 | rev)