Skip to content

Commit e162aac

Browse files
authored
Merge pull request #344 from ggruszczynski/develop
sm_75
2 parents 0f06f69 + 6bdfe5f commit e162aac

File tree

2 files changed

+10
-3
lines changed

2 files changed

+10
-3
lines changed

src/Makefile.in.Rt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -25,7 +25,7 @@ LIBS := @R_LIB@ @LIBS@ @RCPP_LD@ -Xlinker -rpath -Xlinker $(MPI_LIBS) -L$(MPI_LI
2525

2626
TARGETS := CLB.so
2727

28-
NVCC := @CUDA_HOME@/bin/nvcc -gencode arch=compute_10,code=sm_10 -gencode arch=compute_11,code=sm_11 -gencode arch=compute_12,code=sm_12 -gencode arch=compute_13,code=sm_13 -gencode arch=compute_20,code=sm_20 -gencode arch=compute_60,code=sm_60
28+
NVCC := @CUDA_HOME@/bin/nvcc -gencode arch=compute_10,code=sm_10 -gencode arch=compute_11,code=sm_11 -gencode arch=compute_12,code=sm_12 -gencode arch=compute_13,code=sm_13 -gencode arch=compute_20,code=sm_20 -gencode arch=compute_60,code=sm_60, -gencode arch=compute_75,code=sm_75
2929

3030
all: $(TARGETS)
3131

src/configure.ac

Lines changed: 9 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -51,7 +51,7 @@ AC_ARG_ENABLE([cpp11],
5151

5252
AC_ARG_WITH([cuda-arch],
5353
AC_HELP_STRING([--with-cuda-arch=arch],
54-
[specify the desired CUDA architecture (sm_11/sm_13/sm_20/sm_30/sm_60)]))
54+
[specify the desired CUDA architecture (sm_11/sm_13/sm_20/sm_30/sm_60/sm_75)]))
5555

5656
AC_ARG_WITH([nlopt],
5757
AC_HELP_STRING([--with-nlopt=nlopt],
@@ -285,7 +285,14 @@ then
285285
AC_MSG_RESULT([${CUDA_ARCH}])
286286
AC_DEFINE([GRID3D], [1], [Using 3D block grid in CUDA])
287287

288-
CUDA_SUFF="60"
288+
CUDA_SUFF="60"
289+
290+
elif test "$CUDA_ARCH" == "sm_75"
291+
then
292+
AC_MSG_RESULT([${CUDA_ARCH}])
293+
AC_DEFINE([GRID3D], [1], [Using 3D block grid in CUDA])
294+
295+
CUDA_SUFF="75"
289296
else
290297
AC_MSG_ERROR([unknown arch ${CUDA_ARCH}])
291298
fi

0 commit comments

Comments
 (0)