[configure] [nvml] added option to specify nvml header directory path

Description: when compiling with "--enable-nvml" flag we checked for nvml.h only in default locations.
added  "--with-nvml-include-dir=DIR" flag to allow users specifying non-defualt locations where nvml.h might have been installed.

Author: boded

configure.ac

@@ -396,9 +396,26 @@ AC_ARG_ENABLE(nvml,
     [enable_nvml="$default_en_nvml"]
 )
 AC_MSG_RESULT($enable_nvml)
-if test "x$enable_nvml" = "xyes"; then
-    AC_CHECK_HEADER(nvml.h, [], [AC_MSG_ERROR([cannot find nvml.h, please check you nvidia driver installation.])])
-    AC_CHECK_LIB([nvidia-ml], [nvmlInit_v2], [], [AC_MSG_ERROR([libnvidia-ml not found, please check you nvidia driver installation.])])
+
+if test "$enable_nvml" = "yes"; then
+    AC_ARG_WITH([nvml-include-dir], AS_HELP_STRING([--with-nvml-include-dir=DIR], [use nvml from the specified DIR]),
+        [
+            nvml_dir="$withval"
+            AC_MSG_NOTICE([Using nvml from $nvml_dir])
+            CXXFLAGS="$CXXFLAGS -I$nvml_dir"
+            CFLAGS="$CFLAGS -I$nvml_dir"
+        ],
+        [
+            AC_MSG_NOTICE([Looking for nvml headers])
+            PKG_CHECK_MODULES([nvml], [nvidia-ml], [],
+                [
+                    AC_MSG_NOTICE([Looking for nvml headers and libraries in default include paths])
+                    AC_CHECK_HEADERS([nvml.h], [], [AC_MSG_ERROR([cannot find nvml.h in default include paths, please check you nvidia driver installation or use --with-nvml-include-dir=DIR to specify the correct directory path of nvml.h.])])
+                    AC_CHECK_LIB([nvidia-ml], [nvmlInit_v2], [], [AC_MSG_ERROR([libnvidia-ml not found in default library paths, please check you nvidia driver installation])])
+                ]
+            )
+        ]
+    )
     CXXFLAGS="$CXXFLAGS -DENABLE_NVML"
     CFLAGS="$CFLAGS -DENABLE_NVML"
 fi