File tree Expand file tree Collapse file tree 1 file changed +11
-1
lines changed Expand file tree Collapse file tree 1 file changed +11
-1
lines changed Original file line number Diff line number Diff line change @@ -49,6 +49,7 @@ def compile(
49
49
exported_program : ExportedProgram ,
50
50
arg_inputs : Tuple [Any , ...],
51
51
* ,
52
+ inputs : Optional [Tuple [Any , ...]] = None ,
52
53
kwarg_inputs : Optional [dict [Any , Any ]] = None ,
53
54
device : Optional [Union [Device , torch .device , str ]] = _defaults .DEVICE ,
54
55
disable_tf32 : bool = _defaults .DISABLE_TF32 ,
@@ -182,7 +183,16 @@ def compile(
182
183
f"Detected torch_executed_modules was non-empty: { torch_executed_modules } "
183
184
"\n This feature is unimplemented in Torch-TRT Dynamo currently."
184
185
)
185
-
186
+ if inputs :
187
+ logger .warning (
188
+ "'inputs' is deprecated. Please use 'args_inputs' in the future."
189
+ )
190
+ if not arg_inputs :
191
+ arg_inputs = inputs
192
+ else :
193
+ logger .warning (
194
+ "Both 'arg_inputs' and 'inputs' are received. 'inputs' will be ignored."
195
+ )
186
196
if not isinstance (arg_inputs , collections .abc .Sequence ):
187
197
arg_inputs = [arg_inputs ]
188
198
You can’t perform that action at this time.
0 commit comments