# Process this file with autoconf to produce a configure script.

# File: components/nvml/configure.in

AC_INIT

AC_ARG_WITH(nvml_dir,

            [  --with-nvml-dir=<path>   Specify prefix to nvml libraries and headers ],

            [nvml_dir=$withval])

AC_ARG_WITH(nvml_incdir,

           [  --with-nvml-incdir=<path>         Specify directory of nvml header files (nvml.h) in a specific location ],

           [nvml_incdir=$withval])

AC_ARG_WITH(nvml_libdir,

           [  --with-nvml-libdir=<path>         Specify directory of nvml library (libnvidia-ml.so) in a specific location ],

           [nvml_libdir=$withval])

nvml_dotest=0

if test "x$nvml_dir" != "x"; then

    nvml_incdir="$nvml_dir/include"

    nvml_libdir="$nvml_dir/lib64"

    nvml_dotest=1

else

   if test "x$nvml_libdir" != "x"; then

      LIBS="-L$nvml_libdir"

      nvml_dotest=1

    fi

    if test "x$nvml_incdir" != "x"; then

      CPPFLAGS="-I$nvml_incdir"

      nvml_dotest=1

    fi

fi

CFLAGS="$CFLAGS -I$nvml_incdir"

AC_CHECK_HEADER( [nvml.h], [], [AC_MSG_ERROR([NVML component: nvml.h not found: use configure flags to set the path ])], )

LDFLAGS="$LDFLAGS -L$nvml_libdir -Wl,-rpath,$nvml_libdir" 

AC_CHECK_LIB( [nvidia-ml], [nvmlInit], [], [AC_MSG_ERROR([NVML component: libnvidia-ml.so not found: use configure flags to set the path])] )

NVML_INCDIR=$nvml_incdir

NVML_LIBDIR=$nvml_libdir

AC_ARG_WITH(cuda-dir, 

    [--with-cuda-dir=<path> Specify path to cuda root directory], 

    [ CUDA_DIR=$withval

        LDFLAGS="$LDFLAGS -L$withval/lib64 -Wl,-rpath,$withval/lib64"

        AC_CHECK_LIB( [cudart], [cudaMalloc], [], [AC_MSG_ERROR([CUDA cudart library not found!])])],

    [ AC_MSG_ERROR([Component requires path to cuda library.])])

AC_CONFIG_FILES([Makefile.nvml])

AC_SUBST(NVML_INCDIR)

AC_SUBST(NVML_LIBDIR)

AC_SUBST(CUDA_DIR)

AC_OUTPUT