Skip to content

Commit f720276

Browse files
committed
update type for dtype
1 parent 101faf4 commit f720276

File tree

1 file changed

+1
-1
lines changed
  • py/torch_tensorrt/dynamo/conversion/impl/unary

1 file changed

+1
-1
lines changed

py/torch_tensorrt/dynamo/conversion/impl/unary/ops.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -469,7 +469,7 @@ def scalar_tensor(
469469
source_ir: Optional[SourceIR],
470470
name: str,
471471
scalar: Union[int, float, bool],
472-
dtype: Union[torch.int, torch.float, torch.bool] = None,
472+
dtype: Optional[Union[torch.int, torch.float, torch.bool]],
473473
) -> TRTTensor:
474474
tensor = get_trt_tensor(ctx, scalar, f"{name}_scalar_tensor", dtype)
475475
identity_layer = ctx.net.add_identity(tensor)

0 commit comments

Comments
 (0)