yezhengmao/nccl
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Added Pascal nvcc flags, bumped version

Browse Source
This commit is contained in:
Boris Fomitchev 2017-03-24 11:58:14 -07:00
parent 7fef264bfa
commit 649f04d077
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
Makefile
View File

@ -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)