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 024b43b commit 13f1f48Copy full SHA for 13f1f48
py/torch_tensorrt/dynamo/utils.py
@@ -456,7 +456,7 @@ def unwrap_tensor_shape(
456
else:
457
tensor_shape.append((min_max_opt["min"], min_max_opt["max"]))
458
elif isinstance(tensor, torch.SymFloat):
459
- # SymFloats can be an input to graph sometimes. We register their shape as [1] to avoid errors.
+ # SymFloats can be an input to graph sometimes. Although SymFloat is scalar value, we treat it as a 1D tensor throughout Torch-TRT codebase.
460
tensor_shape.append(1)
461
elif isinstance(tensor, (torch.Tensor, FakeTensor)):
462
for dimension in tensor.shape:
0 commit comments