@@ -259,8 +259,25 @@ def run(self):
259
259
package_data = {
260
260
'trtorch' : ['lib/*' ,
261
261
'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' ],
264
281
},
265
282
exclude_package_data = {
266
283
'' : ['*.cpp' ],
0 commit comments