Fix to build with nvcc in path (crux)

This commit is contained in:
Tanguy Pruvot 2014-12-26 15:28:32 +00:00
parent 5cbf239d8a
commit f761531436

View File

@ -147,17 +147,20 @@ AC_ARG_WITH([nvml],
AM_CONDITIONAL([HAVE_NVML], [test -n "$with_nvml"])
NVCC="nvcc"
if test -n "$with_cuda"
then
CUDA_INCLUDES="-I$with_cuda/include"
CUDA_LIBS="-lcudart"
CUDA_LDFLAGS="-L$with_cuda/lib$SUFFIX"
NVCC="$with_cuda/bin/nvcc"
if test -x "$with_cuda/bin/nvcc"
NVCC="$with_cuda/bin/nvcc"
fi
else
CUDA_INCLUDES="-I/usr/local/cuda/include"
CUDA_LIBS="-lcudart -static-libstdc++"
CUDA_LDFLAGS="-L/usr/local/cuda/lib$SUFFIX"
NVCC="nvcc"
fi
if test -n "$with_nvml" ; then