Skip to content

Commit 13f1f48

Browse files
committed
chore: refine comment
Signed-off-by: Dheeraj Peri <[email protected]>
1 parent 024b43b commit 13f1f48

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
@@ -456,7 +456,7 @@ def unwrap_tensor_shape(
456456
else:
457457
tensor_shape.append((min_max_opt["min"], min_max_opt["max"]))
458458
elif isinstance(tensor, torch.SymFloat):
459-
# SymFloats can be an input to graph sometimes. We register their shape as [1] to avoid errors.
459+
# SymFloats can be an input to graph sometimes. Although SymFloat is scalar value, we treat it as a 1D tensor throughout Torch-TRT codebase.
460460
tensor_shape.append(1)
461461
elif isinstance(tensor, (torch.Tensor, FakeTensor)):
462462
for dimension in tensor.shape:

0 commit comments

Comments
 (0)