@@ -106,6 +106,7 @@ def gen_version_file():
106
106
print ("creating version file" )
107
107
f .write ("__version__ = \" " + __version__ + '\" ' )
108
108
109
+
109
110
def copy_libtrtorch (multilinux = False ):
110
111
if not os .path .exists (dir_path + '/trtorch/lib' ):
111
112
os .makedirs (dir_path + '/trtorch/lib' )
@@ -171,15 +172,8 @@ def run(self):
171
172
class CleanCommand (Command ):
172
173
"""Custom clean command to tidy up the project root."""
173
174
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'
183
177
]
184
178
description = "Command to tidy up the project root"
185
179
user_options = []
@@ -265,27 +259,16 @@ def run(self):
265
259
python_requires = '>=3.6' ,
266
260
include_package_data = True ,
267
261
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
+ ],
289
272
},
290
273
exclude_package_data = {
291
274
'' : ['*.cpp' ],
0 commit comments