From 3ab60f5bf95c7af7eaf8110346cc57ccad3155a2 Mon Sep 17 00:00:00 2001
From: Kaipeng Deng <dengkaipeng@baidu.com>
Date: Thu, 21 Nov 2019 15:18:26 +0800
Subject: [PATCH] fix mkldnn include. test=develop (#21247)

* fix mkldnn include. test=develop

* fix mkldnn inlcude. test=develop
---
 python/setup.py.in | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/python/setup.py.in b/python/setup.py.in
index 748880302f..c28aaa1ee3 100644
--- a/python/setup.py.in
+++ b/python/setup.py.in
@@ -286,11 +286,13 @@ headers = (
     list(find_files('*', '${GLOG_INSTALL_DIR}/include')) + # glog
     list(find_files('*', '${BOOST_INCLUDE_DIR}/boost')) + # boost
     list(find_files('*', '${XXHASH_INSTALL_DIR}/include')) + # xxhash
-    list(find_files('mkldnn.h', '${MKLDNN_INSTALL_DIR}/include')) + # mkldnn
     list(find_files('*', '${PROTOBUF_INCLUDE_DIR}')) + # protobuf
     list(find_files('*', '${DLPACK_INCLUDE_DIR}')) + # dlpack 
     list(find_files('*.h', '${THREADPOOL_INCLUDE_DIR}'))) # threadpool
 
+if '${WITH_MKLDNN}' == 'ON':
+    headers += list(find_files('*', '${MKLDNN_INSTALL_DIR}/include')) # mkldnn
+
 
 class InstallCommand(InstallCommandBase):
     def finalize_options(self):
-- 
GitLab