Fix typo in _compile.py (#3128)

juliusgh · web-flow · commit 39f8255020f1 · 2024-08-29T09:18:36.000-07:00
diff --git 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(

Original file line number	Diff line number	Diff line change
`@@ -388,7 +388,7 @@ def convert_method_to_trt_engine(`
`388`	`388`	`torchtrt_kwarg_inputs = prepare_inputs(kwarg_inputs)`
`389`	`389`
`390`	`390`	`exp_program = torch_tensorrt.dynamo.trace(`
`391`		`- module, torchtrt_arg_inputs, kwarg_inputs=torchtrt_kwarg_inputs**kwargs`
	`391`	`+ module, torchtrt_arg_inputs, kwarg_inputs=torchtrt_kwarg_inputs, **kwargs`
`392`	`392`	`)`
`393`	`393`
`394`	`394`	`return dynamo_convert_exported_program_to_serialized_trt_engine(`