Closed
Description
Bug Description
Encountered IndexError
in broadcastable
utility:
File "~/TensorRT/py/torch_tensorrt/dynamo/conversion/converter_utils.py", line 118, in broadcastable
greater_tensor[i] != lesser_tensor[j]
IndexError: tuple index out of range
While executing %where_5 : [num_users=1] = call_function[target=torch.ops.aten.where.self](...)
To Reproduce
Steps to reproduce the behavior:
Invoke aten.where
converter with a 0D tensor.
Expected behavior
Utility should not error.
Environment
- Torch-TensorRT Version (e.g. 1.0.0): 8c62fca
- PyTorch Version (e.g. 1.0):
2.1.0.dev20230803+cu121
Additional context
Blocking compilation of #1697 with torch compile path.