Skip to content

Commit bafa675

Browse files
committed
chore: Add missing header files to the python package
Signed-off-by: Dheeraj Peri <[email protected]>
1 parent ddc0685 commit bafa675

File tree

1 file changed

+19
-2
lines changed

1 file changed

+19
-2
lines changed

py/setup.py

Lines changed: 19 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -259,8 +259,25 @@ def run(self):
259259
package_data={
260260
'trtorch': ['lib/*',
261261
'include/trtorch/*.h',
262-
'include/trtorch/core/**/*.h',
263-
'bin/*'],
262+
'include/trtorch/core/*.h',
263+
'include/trtorch/core/conversion/*.h',
264+
'include/trtorch/core/conversion/conversionctx/*.h',
265+
'include/trtorch/core/conversion/converters/*.h',
266+
'include/trtorch/core/conversion/evaluators/*.h',
267+
'include/trtorch/core/conversion/tensorcontainer/*.h',
268+
'include/trtorch/core/conversion/var/*.h',
269+
'include/trtorch/core/ir/*.h',
270+
'include/trtorch/core/lowering/*.h',
271+
'include/trtorch/core/lowering/passes/*.h',
272+
'include/trtorch/core/partitioning/*.h',
273+
'include/trtorch/core/plugins/*.h',
274+
'include/trtorch/core/plugins/impl/*.h',
275+
'include/trtorch/core/runtime/*.h',
276+
'include/trtorch/core/util/*.h',
277+
'include/trtorch/core/util/logging/*.h',
278+
'bin/*',
279+
'BUILD',
280+
'WORKSPACE'],
264281
},
265282
exclude_package_data={
266283
'': ['*.cpp'],

0 commit comments

Comments
 (0)