Add mkldnn_version.h to pip package (apache#14899)

* add mkldnn_version.h to pip package * add headers from build/install/include folder
haohuanw · Jun 23, 2019 · f2baf50 · f2baf50
1 parent e99064b
commit f2baf50
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tools/pip/setup.py b/tools/pip/setup.py
@@ -150,7 +150,7 @@ def has_ext_modules(self):
         package_data['mxnet'].append('mxnet/libmklml_intel.so')
         package_data['mxnet'].append('mxnet/libiomp5.so')
         package_data['mxnet'].append('mxnet/libmkldnn.so.0')
-    shutil.copytree(os.path.join(CURRENT_DIR, 'mxnet-build/3rdparty/mkldnn/include'),
+    shutil.copytree(os.path.join(CURRENT_DIR, 'mxnet-build/3rdparty/mkldnn/build/install/include'),
                     os.path.join(CURRENT_DIR, 'mxnet/include/mkldnn'))
 if platform.system() == 'Linux':
     shutil.copy(os.path.join(os.path.dirname(LIB_PATH[0]), 'libgfortran.so.3'), os.path.join(CURRENT_DIR, 'mxnet'))