[MXNET-1110] Add header files required by horovod

ci/build_windows.py

@@ -160,9 +160,9 @@ def windows_package(args):
         copy_tree('python', j(pkgdir, 'python'))
         logging.info('packing headers')
         copy_tree('include', j(pkgdir, 'include'))
-        copy_tree(j('3rdparty','dmlc-core','include'), j(pkgdir, 'include'))
-        copy_tree(j('3rdparty','mshadow', 'mshadow'), j(pkgdir, 'include', 'mshadow'))
-        copy_tree(j('3rdparty','tvm','nnvm', 'include'), j(pkgdir,'include', 'nnvm', 'include'))
+        #copy_tree(j('3rdparty','dmlc-core','include'), j(pkgdir, 'include'))
+        #copy_tree(j('3rdparty','mshadow', 'mshadow'), j(pkgdir, 'include', 'mshadow'))
+        #copy_tree(j('3rdparty','tvm','nnvm', 'include'), j(pkgdir,'include', 'nnvm', 'include'))
         logging.info("Compressing package: %s", pkgfile)
         check_call(['7z', 'a', pkgfile, pkgdir])
 

include/dlpack

@@ -0,0 +1 @@
+../3rdparty/dlpack/include/dlpack

include/dmlc

@@ -0,0 +1 @@
+../3rdparty/dmlc-core/include/dmlc

include/mshadow

@@ -0,0 +1 @@
+../3rdparty/mshadow/mshadow

include/nnvm

@@ -0,0 +1 @@
+../3rdparty/tvm/nnvm/include/nnvm

python/mxnet/libinfo.py

@@ -77,5 +77,36 @@ def find_lib_path():
     return lib_path
 
 
+def find_include_path():
+    """Find MXNet dynamic library files.
+
+    Returns
+    -------
+    incl_path : string
+        Path to the header files.
+    """
+    incl_from_env = os.environ.get('MXNET_INCLUDE_PATH')
+    if incl_from_env:
+        if os.path.isfile(incl_from_env):
+            if not os.path.isabs(incl_from_env):
+                logging.warning("MXNET_INCLUDE_PATH should be an absolute path, instead of: %s",
+                                incl_from_env)
+            else:
+                if os.name == 'nt':
+                    os.environ['PATH'] = os.environ['PATH'] + ';' + os.path.dirname(incl_from_env)
+                return [incl_from_env]
+        else:
+            logging.warning("MXNET_INCLUDE_PATH '%s' doesn't exist", incl_from_env)
+
+    curr_path = os.path.dirname(os.path.abspath(os.path.expanduser(__file__)))
+    incl_path = os.path.join(curr_path, '../../include/')
+    if len(incl_path) == 0:
+        raise RuntimeError('Cannot find the MXNet include path.\n')
+
+    if os.name == 'nt':
+        os.environ['PATH'] = os.environ['PATH'] + ';' + os.path.dirname(incl_path)
+    return incl_path
+
+
 # current version
 __version__ = "1.4.0"