Skip to content

Commit d5e3a77

Browse files
committed
docs: Adding words to the refit and engine caching tutorials
Signed-off-by: Naren Dasan <[email protected]>
1 parent f9acc9a commit d5e3a77

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

py/torch_tensorrt/dynamo/utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -347,7 +347,7 @@ def unwrap_tensor_shape(
347347
if isinstance(tensor, int):
348348
tensor_shape.append(tensor)
349349
elif isinstance(tensor, torch.SymInt):
350-
min_max_opt: Dict[str, int] = extract_var_range_info(tensor)
350+
min_max_opt = extract_var_range_info(tensor)
351351
tensor_shape.append((min_max_opt["min"], min_max_opt["max"]))
352352
elif isinstance(tensor, (torch.Tensor, FakeTensor)):
353353
for dimension in tensor.shape:

0 commit comments

Comments
 (0)