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

Better name for GENCODE

Browse Source
This commit is contained in:
Sylvain Jeaugey 2016-06-03 10:22:46 -07:00
parent acb93d1aed
commit 1657af1567
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
Makefile
View File

@ -36,12 +36,12 @@ CUDA_LIB ?= $(CUDA_HOME)/lib64
CUDA_INC ?= $(CUDA_HOME)/include CUDA_INC ?= $(CUDA_HOME)/include
NVCC ?= $(CUDA_HOME)/bin/nvcc NVCC ?= $(CUDA_HOME)/bin/nvcc
CUDACODE ?= -gencode=arch=compute_35,code=sm_35 \ NVCC_GENCODE ?= -gencode=arch=compute_35,code=sm_35 \
-gencode=arch=compute_50,code=sm_50 \ -gencode=arch=compute_50,code=sm_50 \
-gencode=arch=compute_52,code=sm_52 -gencode=arch=compute_52,code=sm_52
CXXFLAGS := -I$(CUDA_INC) -fPIC -fvisibility=hidden CXXFLAGS := -I$(CUDA_INC) -fPIC -fvisibility=hidden
NVCUFLAGS := -ccbin $(CXX) $(CUDACODE) -lineinfo -std=c++11 -maxrregcount 96 NVCUFLAGS := -ccbin $(CXX) $(NVCC_GENCODE) -lineinfo -std=c++11 -maxrregcount 96
# Use addprefix so that we can specify more than one path # Use addprefix so that we can specify more than one path
LDFLAGS := $(addprefix -L,${CUDA_LIB}) -lcudart LDFLAGS := $(addprefix -L,${CUDA_LIB}) -lcudart