From f9cbe3bdcc1c824da136890d84fbdc7e9119dcce Mon Sep 17 00:00:00 2001 From: Kaipeng Deng Date: Sat, 23 Nov 2019 15:09:39 +0800 Subject: [PATCH] add mkldnn include. test=develop (#21314) --- python/setup.py.in | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/python/setup.py.in b/python/setup.py.in index 6b0eecb21d4..35033683a80 100644 --- a/python/setup.py.in +++ b/python/setup.py.in @@ -271,9 +271,11 @@ def find_files(pattern, root): headers = ( list(find_files('*.h', '@PADDLE_SOURCE_DIR@/paddle/fluid/framework')) + + list(find_files('*.h', '@PADDLE_SOURCE_DIR@/paddle/fluid/imperative')) + list(find_files('*.h', '@PADDLE_SOURCE_DIR@/paddle/fluid/memory')) + list(find_files('*.h', '@PADDLE_SOURCE_DIR@/paddle/fluid/platform')) + list(find_files('*.h', '@PADDLE_SOURCE_DIR@/paddle/fluid/string')) + + list(find_files('*.pb.h', '${PADDLE_BINARY_DIR}/paddle/fluid/platform')) + list(find_files('*.pb.h', '${PADDLE_BINARY_DIR}/paddle/fluid/framework')) + ['${EIGEN_INCLUDE_DIR}/Eigen/Core'] + # eigen list(find_files('*', '${EIGEN_INCLUDE_DIR}/Eigen/src')) + # eigen @@ -286,6 +288,9 @@ headers = ( 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): @@ -333,6 +338,7 @@ class InstallHeaders(Command): 'install/protobuf/include', 'install/gflags/include', 'install/glog/include', 'install/xxhash/include', + 'install/mkldnn/include', 'threadpool/src/extern_threadpool'] for pattern in patterns: install_dir = re.sub(pattern, '', install_dir) -- GitLab