[incubator-mxnet] branch master updated: Update tools/pip/setup.py for oneDNN v2 (#19694)

[sx...@apache.org]

This is an automated email from the ASF dual-hosted git repository.

sxjscience pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/incubator-mxnet.git


The following commit(s) were added to refs/heads/master by this push:
     new 9c1448f  Update tools/pip/setup.py for oneDNN v2 (#19694)
9c1448f is described below

commit 9c1448fedb436c82e13b8e164043f56ef5ba4a9b
Author: Leonard Lausen <la...@amazon.com>
AuthorDate: Fri Dec 18 20:59:34 2020 -0600

    Update tools/pip/setup.py for oneDNN v2 (#19694)
---
 tools/pip/setup.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tools/pip/setup.py b/tools/pip/setup.py
index 97602e5..f3a7443 100644
--- a/tools/pip/setup.py
+++ b/tools/pip/setup.py
@@ -146,7 +146,7 @@ short_description += ' This version uses {0}.'.format(' and '.join(libraries))
 package_data = {'mxnet': [os.path.join('mxnet', os.path.basename(LIB_PATH[0]))],
                 'dmlc_tracker': []}
 if Features().is_enabled("MKLDNN"):
-    shutil.copytree(os.path.join(CURRENT_DIR, 'mxnet-build/3rdparty/mkldnn/include'),
+    shutil.copytree(os.path.join(CURRENT_DIR, 'mxnet-build/3rdparty/mkldnn/include/oneapi/dnnl'),
                     os.path.join(CURRENT_DIR, 'mxnet/include/mkldnn'))
 if platform.system() == 'Linux':
     libdir, mxdir = os.path.dirname(LIB_PATH[0]), os.path.join(CURRENT_DIR, 'mxnet')