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 ,
@@ -168,7 +169,16 @@ def compile(
168
169
f"Detected torch_executed_modules was non-empty: { torch_executed_modules } "
169
170
"\n This feature is unimplemented in Torch-TRT Dynamo currently."
170
171
)
171
-
172
+ if inputs :
173
+ logger .warning (
174
+ "'inputs' is deprecated. Please use 'args_inputs' in the future."
175
+ )
176
+ if not arg_inputs :
177
+ arg_inputs = inputs
178
+ else :
179
+ logger .warning (
180
+ "Both 'arg_inputs' and 'inputs' are received. 'inputs' will be ignored."
181
+ )
172
182
if not isinstance (arg_inputs , collections .abc .Sequence ):
173
183
arg_inputs = [arg_inputs ]
174
184
You can’t perform that action at this time.
0 commit comments