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 <[email protected]>
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')
