Skip to content

Commit 732c666

Browse files
authored
Update ax_check_cuda.m4
1 parent 8a82cfe commit 732c666

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

build-aux/m4/ax_check_cuda.m4

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -51,9 +51,10 @@ then
5151
CUDA_CFLAGS+=" -I$with_cuda/include"
5252
CUDA_LDLIBS+=" -L$with_cuda/lib64"
5353
else
54-
#AC_CHECK_FILE(/usr/local/cuda/,[CUDAPATH="/usr/local/cuda"],[])
55-
#AC_CHECK_FILE(/usr/local/cuda/include,[CUDA_CFLAGS+=" -I/usr/local/cuda/include"],[CUDA_CFLAGS=""])
56-
#AC_CHECK_FILE(/usr/local/cuda/lib64,[CUDA_LDLIBS+=" -L/usr/local/cuda/lib64"],[])
54+
AX_NORMALIZE_PATH([with_cuda], ["/"])
55+
CUDAPATH="$with_cuda"
56+
CUDA_CFLAGS+=" -I$with_cuda/include"
57+
CUDA_LDLIBS+=" -L$with_cuda/lib64"
5758
fi
5859
CUDA_LDLIBS+=" -lcuda -lcudart -lcublas"
5960

0 commit comments

Comments
 (0)