Skip to content

Commit 61a5421

Browse files
committed
fix: fix signature mismatches issue for non-tensor input in plugin converter
1 parent 530a2cd commit 61a5421

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

py/torch_tensorrt/dynamo/conversion/plugins/_generate_plugin_converter.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -89,9 +89,8 @@ def custom_kernel_converter(
8989
non_tensor_args = args[len(tensor_inputs) :]
9090
non_tensor_kwargs = dict(zip(list(non_tensor_inputs.keys()), non_tensor_args))
9191

92-
for k, v in kwargs.items():
93-
if k in non_tensor_kwargs:
94-
kwargs[k] = non_tensor_kwargs[k]
92+
# Update kwargs with non_tensor_kwargs, adding new keys or overwriting existing ones
93+
kwargs.update(non_tensor_kwargs)
9594

9695
for k, v in kwargs.items():
9796
if isinstance(v, torch.fx.immutable_collections.immutable_list):

0 commit comments

Comments
 (0)