fix mkldnn include. test=develop (#21247)

* fix mkldnn include. test=develop * fix mkldnn inlcude. test=develop

fix mkldnn include. test=develop (#21247)
* fix mkldnn include. test=develop * fix mkldnn inlcude. test=develop
3ab60f5b · Kaipeng Deng · GitHub · eca66f31 · 3ab60f5b
隐藏空白更改
内联并排

Showing with 3 addition and 1 deletion

python/setup.py.in python/setup.py.in +3 -1

未找到文件。
--- 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):