-
Notifications
You must be signed in to change notification settings - Fork 7
/
Copy pathconfigure.ac
148 lines (115 loc) · 4.32 KB
/
configure.ac
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
AC_INIT([gpuRcuda], 1.0.0)
AC_LANG(C++)
AC_REQUIRE_CPP
echo "Checking for C++ Compiler"
AC_PROG_CXX
GPURCUDA_CPPFLAGS="-I. -I../inst/include"
GPURCUDA_CXXFLAGS=""
GPURCUDA_LIBS=""
if test "$BACKEND" = "OPENMP"; then
AC_MSG_NOTICE("BACKEND = OPENMP")
GPURCUDA_CPPFLAGS="${GPURCUDA_CPPFLAGS} -DTHRUST_DEVICE_SYSTEM=THRUST_DEVICE_SYSTEM_OMP"
GPURCUDA_CXXFLAGS="${GPURCUDA_CXXFLAGS} -fopenmp"
BACKEND="OPENMP"
AC_SUBST(BACKEND)
elif test "$BACKEND" = "CUDA"; then
AC_MSG_NOTICE("BACKEND = CUDA")
GPURCUDA_CPPFLAGS="${GPURCUDA_CPPFLAGS} -DTHRUST_DEVICE_SYSTEM=THRUST_DEVICE_SYSTEM_CUDA"
AC_MSG_CHECKING("Checking environment variable CUDA_HOME")
if test -z "${CUDA_HOME}"; then
CUDA_HOME=`find /usr/local/ -maxdepth 1 -type d -name "cuda*" | sort -V | tail -1`
AC_MSG_RESULT("CUDA_HOME not set; using highest version found ${CUDA_HOME}")
else
AC_MSG_RESULT("using CUDA_HOME=${CUDA_HOME}")
fi
AC_CHECK_FILE([${CUDA_HOME}/bin/nvcc], [HAS_NVCC="yes"])
if test -z "${HAS_NVCC}"; then
AC_MSG_ERROR("NVCC not found; check CUDA install")
else
AC_MSG_RESULT("NVCC found")
fi
AC_MSG_CHECKING("whether this is the 64 bit linux version of CUDA")
AC_CHECK_FILE([${CUDA_HOME}/lib64/libcudart.so], [BIT64="yes"])
if test -z "${BIT64}"; then
AC_MSG_RESULT("no -- using ${CUDA_HOME}/lib for CUDA libs")
cu_libdir="${CUDA_HOME}/lib"
#AC_SUBST(CUDA_LIBDIR,["$cu_libdir"])
else
AC_MSG_RESULT("yes -- using ${CUDA_HOME}/lib64 for CUDA libs")
cu_libdir="${CUDA_HOME}/lib64"
#AC_SUBST(CUDA_LIBDIR,["$cu_libdir"])
fi
GPURCUDA_LIBS="${GPURCUDA_LIBS} -L${cu_libdir} -Wl,-rpath,${cu_libdir} -lcudart"
BACKEND="CUDA"
AC_SUBST(BACKEND)
else
AC_MSG_NOTICE("No recognized BACKEND defined so trying CUDA first")
AC_MSG_CHECKING("Checking environment variable CUDA_HOME")
if test -z "${CUDA_HOME}"; then
CUDA_HOME=`find /usr/local/ -maxdepth 1 -type d -name "cuda*" | sort -V | tail -1`
fi
#else
# AC_MSG_RESULT("using CUDA_HOME=${CUDA_HOME}")
#fi
if test -z "${CUDA_HOME}"; then
AC_MSG_RESULT("No CUDA installation found, attempting OPENMP install")
GPURCUDA_CPPFLAGS="${GPURCUDA_CPPFLAGS} -DTHRUST_DEVICE_SYSTEM=THRUST_DEVICE_SYSTEM_OMP"
GPURCUDA_CXXFLAGS="${CXXFLAGS} -fopenmp"
BACKEND="OPENMP"
AC_SUBST(BACKEND)
else
AC_MSG_RESULT("using CUDA_HOME=${CUDA_HOME}")
#AC_MSG_RESULT("CUDA_HOME not set; using highest version found ${CUDA_HOME}")
AC_CHECK_FILE([${CUDA_HOME}/bin/nvcc], [HAS_NVCC="yes"])
if test -z "${HAS_NVCC}"; then
AC_MSG_ERROR("NVCC not found; check CUDA install")
else
AC_MSG_RESULT("NVCC found")
fi
AC_MSG_CHECKING("whether this is the 64 bit linux version of CUDA")
AC_CHECK_FILE([${CUDA_HOME}/lib64/libcudart.so], [BIT64="yes"])
if test -z "${BIT64}"; then
AC_MSG_RESULT("no -- using ${CUDA_HOME}/lib for CUDA libs")
cu_libdir="${CUDA_HOME}/lib"
#AC_SUBST(CUDA_LIBDIR,["$cu_libdir"])
else
AC_MSG_RESULT("yes -- using ${CUDA_HOME}/lib64 for CUDA libs")
cu_libdir="${CUDA_HOME}/lib64"
#AC_SUBST(CUDA_LIBDIR,["$cu_libdir"])
fi
GPURCUDA_LIBS="${GPURCUDA_LIBS} -L${cu_libdir} -Wl,-rpath,${cu_libdir} -lcudart"
BACKEND="CUDA"
AC_SUBST(BACKEND)
fi
fi
## look for Rscript, but use the one found via R_HOME to allow for multiple installations
AC_DEFUN([AC_PROG_RSCRIPT], [AC_CHECK_PROG(RSCRIPT,Rscript,yes)])
AC_PROG_RSCRIPT
if test x"${RSCRIPT}" == x"yes" ; then
R_HOME=`Rscript -e 'cat(R.home())'`
rcpp_incl=`${R_HOME}/bin/Rscript -e 'Rcpp:::CxxFlags()'`
thrust_incl=`${R_HOME}/bin/Rscript -e 'cat(paste("-I", system.file("include", package = "thrust"), sep = ""))'`
AC_SUBST(RCPP_INCL,["$rcpp_incl"])
AC_SUBST(THRUST_INCL,["$thrust_incl"])
else
echo "
Your installation does not appear to have Rscript installed.
Please make sure that you have a working and complete R installation.
"
exit 1
fi
R_INCL=`"${R_HOME}/bin/R" CMD config --cppflags`
R_LIBS=`"${R_HOME}/bin/R" CMD config --ldflags`
R_CPIC=`"${R_HOME}/bin/R" CMD config CPICFLAGS`
AC_MSG_CHECKING("building the nvcc command line")
NVCC="${CUDA_HOME}/bin/nvcc"
AC_MSG_NOTICE([Building Makevars])
AC_SUBST(GPURCUDA_CPPFLAGS)
AC_SUBST(GPURCUDA_CXXFLAGS)
AC_SUBST(CU_INCL)
AC_SUBST(NVCC)
AC_SUBST(R_INCL)
AC_SUBST(R_CPIC)
AC_SUBST(R_LIBS)
AC_SUBST(GPURCUDA_LIBS)
AC_OUTPUT(src/Makevars)