Skip to content

Commit 28f871d

Browse files
authored
[cherry-pick] add dlpack include in third_party (#20739)
* add dlpack include in third_party. test=develop
1 parent 06452f1 commit 28f871d

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

python/setup.py.in

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -283,6 +283,7 @@ headers = (
283283
list(find_files('*', '${BOOST_INCLUDE_DIR}/boost')) + # boost
284284
list(find_files('*', '${XXHASH_INSTALL_DIR}/include')) + # xxhash
285285
list(find_files('*', '${PROTOBUF_INCLUDE_DIR}')) + # protobuf
286+
list(find_files('*', '${DLPACK_INCLUDE_DIR}')) + # dlpack
286287
list(find_files('*.h', '${THREADPOOL_INCLUDE_DIR}'))) # threadpool
287288

288289

@@ -329,6 +330,7 @@ class InstallHeaders(Command):
329330
install_dir = re.sub('${THIRD_PARTY_PATH}', 'third_party', header)
330331
install_dir = re.sub('src/extern_eigen3/', '', install_dir)
331332
install_dir = re.sub('src/extern_boost/', '', install_dir)
333+
install_dir = re.sub('src/extern_dlpack/', '', install_dir)
332334
install_dir = os.path.join(self.install_dir, os.path.dirname(install_dir))
333335
if not os.path.exists(install_dir):
334336
self.mkpath(install_dir)

0 commit comments

Comments
 (0)