From 39f8255020f108956d5c93a08e4cf4c1c32fbac7 Mon Sep 17 00:00:00 2001 From: Julius Herb Date: Thu, 29 Aug 2024 18:18:36 +0200 Subject: [PATCH] Fix typo in _compile.py (#3128) --- py/torch_tensorrt/_compile.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/py/torch_tensorrt/_compile.py b/py/torch_tensorrt/_compile.py index 910e98fa41..cd49962c17 100644 --- a/py/torch_tensorrt/_compile.py +++ b/py/torch_tensorrt/_compile.py @@ -388,7 +388,7 @@ def convert_method_to_trt_engine( torchtrt_kwarg_inputs = prepare_inputs(kwarg_inputs) exp_program = torch_tensorrt.dynamo.trace( - module, torchtrt_arg_inputs, kwarg_inputs=torchtrt_kwarg_inputs**kwargs + module, torchtrt_arg_inputs, kwarg_inputs=torchtrt_kwarg_inputs, **kwargs ) return dynamo_convert_exported_program_to_serialized_trt_engine(