This is an automated email from the ASF dual-hosted git repository. lausen pushed a commit to branch leezu-patch-4 in repository https://gitbox.apache.org/repos/asf/incubator-mxnet.git
commit 2d5c23cea3eb3b9380156b303378356fe1849f28 Author: Leonard Lausen <[email protected]> AuthorDate: Wed May 13 17:49:20 2020 -0700 Fix missing MKLDNN headers --- 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 7aaebf5..e6464da 100644 --- a/tools/pip/setup.py +++ b/tools/pip/setup.py @@ -149,7 +149,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 variant.endswith('MKL'): +if variant != 'native': 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':
