Skip to content

Commit 3375f49

Browse files
committed
Added support for bool input
1 parent 76437f9 commit 3375f49

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

py/torch_tensorrt/dynamo/runtime/_MutableTorchTensorRTModule.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -442,9 +442,9 @@ def _process_kwarg_inputs(inputs: Any) -> Any:
442442
return {
443443
k: MutableTorchTensorRTModule._process_kwarg_inputs(v)
444444
for k, v in inputs.items()
445-
if (v is not None and not isinstance(v, bool))
445+
if (v is not None)
446446
}
447-
elif isinstance(inputs, torch.Tensor):
447+
elif isinstance(inputs, (torch.Tensor, bool)):
448448
return inputs
449449
elif isinstance(inputs, (int, float, np.ndarray)):
450450
return torch.tensor(inputs)

0 commit comments

Comments
 (0)