diff --git a/m4/ax_cuda.m4 b/m4/ax_cuda.m4 index da7dc73fe9a41cc1aa81ddc2ad42fb2f1b28e8a0..cdf6e17f76f7541a733fa277b5ae3dfbd3311442 100644 --- a/m4/ax_cuda.m4 +++ b/m4/ax_cuda.m4 @@ -57,8 +57,12 @@ AS_IF([test "x$NVCC_EXIST" = "xno"],[],[ # If $build_cpu contains "_64", append "64" to CUDA_LIBS AS_IF([echo $build_cpu | grep -q "_64"], [ - AS_IF([ ! test -d {CUDA_LIBS}lib64 ], [ CUDA_LIBS+="64" ]) - AS_IF([ ! command -v bumblebee >/dev/null ], [ CUDA_LIBS+=" -L/usr/lib64/nvidia-bumblebee/ " ]) + AS_IF([ test -d {CUDA_LIBS}lib64 ], [ CUDA_LIBS+="64" ]) + AS_IF([ command -v bumblebee >/dev/null ], [ CUDA_LIBS+=" -L/usr/lib64/nvidia-bumblebee/ " ]) + AS_IF([ test -d /usr/local/cuda/lib64 ], [ CUDA_LIBS+=" -L/usr/local/cuda/lib64 " ], + [ + AS_IF([ test -d /usr/local/cuda/lib ],[ CUDA_LIBS+=" -L/usr/local/cuda/lib " ]) + ]) ]) # Append " -lcuda -lcudart" to CUDA_LIBS