We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 530a2cd commit 61a5421Copy full SHA for 61a5421
py/torch_tensorrt/dynamo/conversion/plugins/_generate_plugin_converter.py
@@ -89,9 +89,8 @@ def custom_kernel_converter(
89
non_tensor_args = args[len(tensor_inputs) :]
90
non_tensor_kwargs = dict(zip(list(non_tensor_inputs.keys()), non_tensor_args))
91
92
- for k, v in kwargs.items():
93
- if k in non_tensor_kwargs:
94
- kwargs[k] = non_tensor_kwargs[k]
+ # Update kwargs with non_tensor_kwargs, adding new keys or overwriting existing ones
+ kwargs.update(non_tensor_kwargs)
95
96
for k, v in kwargs.items():
97
if isinstance(v, torch.fx.immutable_collections.immutable_list):
0 commit comments