From 805f0e4542d78ab0d9d3160757020a93bf8acd82 Mon Sep 17 00:00:00 2001
From: bartekkuncer <bartosz.kuncer@intel.com>
Date: Tue, 5 Jan 2021 14:44:48 +0100
Subject: [PATCH] Fix tools/pip/setup.py (#19726)

---
 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 80624c9b46ae..7d336b072f33 100644
--- a/tools/pip/setup.py
+++ b/tools/pip/setup.py
@@ -150,7 +150,7 @@ def skip_markdown_comments(md):
 
 package_data = {'mxnet': [os.path.join('mxnet', os.path.basename(LIB_PATH[0]))],
                 'dmlc_tracker': []}
-if variant != 'NATIVE':
+if Features().is_enabled("MKLDNN"):
     shutil.copytree(os.path.join(CURRENT_DIR, 'mxnet-build/3rdparty/mkldnn/include'),
                     os.path.join(CURRENT_DIR, 'mxnet/include/mkldnn'))
 if platform.system() == 'Linux':