Fix missing MKLDNN headers (apache#18310)

AntiZpvoh · Jul 6, 2020 · e4a5b6e · e4a5b6e
1 parent c36f3df
commit e4a5b6e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tools/pip/setup.py b/tools/pip/setup.py
@@ -149,7 +149,7 @@ def skip_markdown_comments(md):
 
 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':