Skip to content

Commit 211ca8f

Browse files
committed
remove unused args in interpret_module_to_result()
1 parent 2773276 commit 211ca8f

File tree

2 files changed

+2
-9
lines changed

2 files changed

+2
-9
lines changed

py/torch_tensorrt/dynamo/_compiler.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1339,15 +1339,13 @@ def convert_exported_program_to_serialized_trt_engine(
13391339
)
13401340

13411341
flattened_input_list = get_flat_args_with_check(
1342-
exported_program, list(trt_arg_inputs), trt_kwarg_inputs # type: ignore
1342+
exported_program, list(trt_arg_inputs), trt_kwarg_inputs
13431343
)[0]
13441344

13451345
try:
13461346
interpreter_result = interpret_module_to_result(
13471347
gm,
13481348
inputs=flattened_input_list,
1349-
arg_inputs=list(trt_arg_inputs),
1350-
kwarg_inputs=trt_kwarg_inputs,
13511349
settings=settings,
13521350
engine_cache=engine_cache,
13531351
)

py/torch_tensorrt/dynamo/conversion/_conversion.py

Lines changed: 1 addition & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -179,17 +179,12 @@ def interpret_module_to_result(
179179
module: torch.fx.GraphModule,
180180
inputs: Sequence[Input],
181181
settings: CompilationSettings = CompilationSettings(),
182-
arg_inputs: Optional[Sequence[Input]] = None,
183-
kwarg_inputs: Optional[dict[str, Any]] = None,
184182
engine_cache: Optional[BaseEngineCache] = None,
185183
) -> SerializedInterpreterResult:
186184
"""Interpret an FX module to a TRTInterpreterResult
187185
Args:
188186
module: FX GraphModule to interpret
189-
inputs: Sequence of FLATTENED Tensors representing inputs to the module. It should include both
190-
arg_inputs and kwarg_inputs, if applicable.
191-
arg_inputs: Sequence of Tensors representing inputs to the module.
192-
kwarg_inputs: A dictionary of Tensors representing inputs to the module.
187+
inputs: It requires a sequence of FLATTENED Inputs representing inputs to the module. It should include both arg_inputs and kwarg_inputs, if applicable.
193188
settings: Compilation settings
194189
engine_cache: Engine cache instance
195190
Returns:

0 commit comments

Comments
 (0)