From ba44f654d6b612f4619a0afa7351aebaf7e50577 Mon Sep 17 00:00:00 2001 From: Kaipeng Deng Date: Sun, 20 Oct 2019 22:32:18 +0800 Subject: [PATCH] add dlpack include in third_party. test=develop (#20737) * add dlpack include in third_party. test=develop --- python/setup.py.in | 2 ++ 1 file changed, 2 insertions(+) diff --git a/python/setup.py.in b/python/setup.py.in index 6a97d65d3c1..9c786525250 100644 --- a/python/setup.py.in +++ b/python/setup.py.in @@ -283,6 +283,7 @@ headers = ( list(find_files('*', '${BOOST_INCLUDE_DIR}/boost')) + # boost list(find_files('*', '${XXHASH_INSTALL_DIR}/include')) + # xxhash list(find_files('*', '${PROTOBUF_INCLUDE_DIR}')) + # protobuf + list(find_files('*', '${DLPACK_INCLUDE_DIR}')) + # dlpack list(find_files('*.h', '${THREADPOOL_INCLUDE_DIR}'))) # threadpool @@ -329,6 +330,7 @@ class InstallHeaders(Command): install_dir = re.sub('${THIRD_PARTY_PATH}', 'third_party', header) install_dir = re.sub('src/extern_eigen3/', '', install_dir) install_dir = re.sub('src/extern_boost/', '', install_dir) + install_dir = re.sub('src/extern_dlpack/', '', install_dir) install_dir = os.path.join(self.install_dir, os.path.dirname(install_dir)) if not os.path.exists(install_dir): self.mkpath(install_dir) -- GitLab