Skip to content

Commit

Permalink
Upgrade compute capability
Browse files Browse the repository at this point in the history
I was having problems with Thrust on my Ubuntu
box with CUDA 8.0. Then I found BVLC/caffe#138,
which contained the solution. Apparently, I just needed
to [upgrade the compute capability](BVLC/caffe#138 (comment)).
  • Loading branch information
wlandau-lilly committed Sep 30, 2017
1 parent 25db2ea commit 58c4559
Show file tree
Hide file tree
Showing 2 changed files with 6 additions and 2 deletions.
2 changes: 1 addition & 1 deletion DESCRIPTION
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,7 @@ Package: fbseqCUDA
Title: fbseqCUDA
Description: Internal CUDA code for package fbseq.
URL: https://github.com/wlandau/fbseqCUDA
Version: 0.0
Version: 0.1
Author: Will Landau <[email protected]>
Maintainer: Will Landau <[email protected]>
Depends:
Expand Down
6 changes: 5 additions & 1 deletion src/Makevars
Original file line number Diff line number Diff line change
Expand Up @@ -6,7 +6,11 @@ R_INC = ${R_HOME}/include
NVCC = ${CUDA_HOME}/bin/nvcc
CUDA_INC = ${CUDA_HOME}/include
PKG_LIBS = -L${CUDA_HOME}/lib64 -Wl,-rpath,${CUDA_HOME}/lib64 -lcuda -lcudart
NVCC_FLAGS = -x cu -I${R_INC} -Xcompiler "-fPIC" -gencode arch=compute_20,code=sm_20 -gencode arch=compute_30,code=sm_30 -gencode arch=compute_35,code=sm_35
NVCC_FLAGS = --machine 64 -x cu -I${R_INC} -Xcompiler "-fPIC" \
-gencode arch=compute_30,code=sm_30 \
-gencode arch=compute_35,code=sm_35 \
-gencode arch=compute_50,code=sm_50 \
-gencode arch=compute_50,code=compute_50

OBJECTS= fbseqCUDA.o

Expand Down

0 comments on commit 58c4559

Please sign in to comment.