@@ -106,6 +106,7 @@ def gen_version_file():
106106 print ("creating version file" )
107107 f .write ("__version__ = \" " + __version__ + '\" ' )
108108
109+
109110def copy_libtrtorch (multilinux = False ):
110111 if not os .path .exists (dir_path + '/trtorch/lib' ):
111112 os .makedirs (dir_path + '/trtorch/lib' )
@@ -171,15 +172,8 @@ def run(self):
171172class CleanCommand (Command ):
172173 """Custom clean command to tidy up the project root."""
173174 PY_CLEAN_FILES = [
174- './build' ,
175- './dist' ,
176- './trtorch/__pycache__' ,
177- './trtorch/lib' ,
178- './trtorch/include' ,
179- './trtorch/bin' ,
180- './*.pyc' ,
181- './*.tgz' ,
182- './*.egg-info'
175+ './build' , './dist' , './trtorch/__pycache__' , './trtorch/lib' , './trtorch/include' , './trtorch/bin' , './*.pyc' ,
176+ './*.tgz' , './*.egg-info'
183177 ]
184178 description = "Command to tidy up the project root"
185179 user_options = []
@@ -265,27 +259,16 @@ def run(self):
265259 python_requires = '>=3.6' ,
266260 include_package_data = True ,
267261 package_data = {
268- 'trtorch' : ['lib/*' ,
269- 'include/trtorch/*.h' ,
270- 'include/trtorch/core/*.h' ,
271- 'include/trtorch/core/conversion/*.h' ,
272- 'include/trtorch/core/conversion/conversionctx/*.h' ,
273- 'include/trtorch/core/conversion/converters/*.h' ,
274- 'include/trtorch/core/conversion/evaluators/*.h' ,
275- 'include/trtorch/core/conversion/tensorcontainer/*.h' ,
276- 'include/trtorch/core/conversion/var/*.h' ,
277- 'include/trtorch/core/ir/*.h' ,
278- 'include/trtorch/core/lowering/*.h' ,
279- 'include/trtorch/core/lowering/passes/*.h' ,
280- 'include/trtorch/core/partitioning/*.h' ,
281- 'include/trtorch/core/plugins/*.h' ,
282- 'include/trtorch/core/plugins/impl/*.h' ,
283- 'include/trtorch/core/runtime/*.h' ,
284- 'include/trtorch/core/util/*.h' ,
285- 'include/trtorch/core/util/logging/*.h' ,
286- 'bin/*' ,
287- 'BUILD' ,
288- 'WORKSPACE' ],
262+ 'trtorch' : [
263+ 'lib/*' , 'include/trtorch/*.h' , 'include/trtorch/core/*.h' , 'include/trtorch/core/conversion/*.h' ,
264+ 'include/trtorch/core/conversion/conversionctx/*.h' , 'include/trtorch/core/conversion/converters/*.h' ,
265+ 'include/trtorch/core/conversion/evaluators/*.h' , 'include/trtorch/core/conversion/tensorcontainer/*.h' ,
266+ 'include/trtorch/core/conversion/var/*.h' , 'include/trtorch/core/ir/*.h' ,
267+ 'include/trtorch/core/lowering/*.h' , 'include/trtorch/core/lowering/passes/*.h' ,
268+ 'include/trtorch/core/partitioning/*.h' , 'include/trtorch/core/plugins/*.h' ,
269+ 'include/trtorch/core/plugins/impl/*.h' , 'include/trtorch/core/runtime/*.h' ,
270+ 'include/trtorch/core/util/*.h' , 'include/trtorch/core/util/logging/*.h' , 'bin/*' , 'BUILD' , 'WORKSPACE'
271+ ],
289272 },
290273 exclude_package_data = {
291274 '' : ['*.cpp' ],
0 commit comments