AC_DEFUN([ASTRA_CHECK_BOOST_THREAD],[ BOOST_BACKUP_LIBS="$LIBS" LIBS="$LIBS $1" AC_LINK_IFELSE([AC_LANG_SOURCE([ #include int main() { boost::thread t; boost::posix_time::milliseconds m(1); t.timed_join(m); return 0; } ])],[$2],[$3]) LIBS="$BOOST_BACKUP_LIBS" unset BOOST_BACKUP_LIBS ]) AC_DEFUN([ASTRA_CHECK_BOOST_UNIT_TEST_FRAMEWORK],[ BOOST_BACKUP_LIBS="$LIBS" LIBS="$LIBS $1" AC_LINK_IFELSE([AC_LANG_SOURCE([ #define BOOST_TEST_DYN_LINK #define BOOST_AUTO_TEST_MAIN #include #include #include ])],[$2],[$3]) LIBS="$BOOST_BACKUP_LIBS" unset BOOST_BACKUP_LIBS ]) dnl ASTRA_CHECK_MEX_SUFFIX(list-of-suffices, variable-to-set) AC_DEFUN([ASTRA_CHECK_MEX_SUFFIX],[ cat >conftest.cc <<_ACEOF extern "C" void mexFunction() { } _ACEOF $CXX -fPIC -c -o conftest.o conftest.cc $MEX -cxx -output conftest conftest.o $2="" for suffix in $1; do if test -f "conftest.$suffix"; then $2="$suffix" rm -f "conftest.$suffix" fi done rm -f conftest.cc conftest.o ]) dnl ASTRA_RUN_STOREOUTPUT(command, output) AC_DEFUN([ASTRA_RUN_STOREOUTPUT],[{ AS_ECHO(["$as_me:${as_lineno-$LINENO}: $1"]) >&AS_MESSAGE_LOG_FD ( $1 ) >$2 2>&1 ac_status=$? cat $2 >&AS_MESSAGE_LOG_FD AS_ECHO(["$as_me:${as_lineno-$LINENO}: \$? = $ac_status"]) >&AS_MESSAGE_LOG_FD test $ac_status = 0; }]) dnl ASTRA_RUN(command) AC_DEFUN([ASTRA_RUN],[ASTRA_RUN_STOREOUTPUT($1,/dev/null)]) dnl ASTRA_CHECK_NVCC(variable-to-set,cppflags-to-set) AC_DEFUN([ASTRA_CHECK_NVCC],[ cat >conftest.cu <<_ACEOF #include int main() { std::cout << "Test" << std::endl; return 0; } _ACEOF $1="yes" ASTRA_RUN_STOREOUTPUT([$NVCC -c -o conftest.o conftest.cu $$2],conftest.nvcc.out) || { $1="no" # Check if hack for gcc 4.4 helps if grep -q __builtin_stdarg_start conftest.nvcc.out; then NVCC_OPT="-Xcompiler -D__builtin_stdarg_start=__builtin_va_start" ASTRA_RUN([$NVCC -c -o conftest.o conftest.cu $$2 $NVCC_OPT]) && { $1="yes" $2="$$2 $NVCC_OPT" } fi } if test x$$1 = xno; then AS_ECHO(["$as_me: failed program was:"]) >&AS_MESSAGE_LOG_FD sed 's/^/| /' conftest.cu >&AS_MESSAGE_LOG_FD fi rm -f conftest.cu conftest.o conftest.nvcc.out ]) dnl ASTRA_FIND_NVCC_ARCHS(archs-to-try,cppflags-to-extend,output-list) dnl Architectures should be of the form 10,20,30,35, dnl and should be in order. The last accepted one will be used for PTX output. dnl All accepted ones will be used for cubin output. AC_DEFUN([ASTRA_FIND_NVCC_ARCHS],[ cat >conftest.cu <<_ACEOF #include int main() { std::cout << "Test" << std::endl; return 0; } _ACEOF NVCC_lastarch="none" NVCC_extra="" NVCC_list="" astra_save_IFS=$IFS IFS=, for arch in $1; do IFS=$astra_save_IFS NVCC_opt="-gencode=arch=compute_$arch,code=sm_$arch" $NVCC -c -o conftest.o conftest.cu $$2 $NVCC_opt >conftest.nvcc.out 2>&1 && { NVCC_lastarch=$arch NVCC_extra="$NVCC_extra $NVCC_opt" NVCC_list="${NVCC_list:+$NVCC_list, }$arch" } done IFS=$astra_save_IFS if test $NVCC_lastarch != none; then NVCC_extra="$NVCC_extra -gencode=arch=compute_${NVCC_lastarch},code=compute_${NVCC_lastarch}" $3="$NVCC_list" $2="$$2 $NVCC_extra" else $3="none" fi rm -f conftest.cu conftest.o conftest.nvcc.out ])