Skip to content

🐛 [Bug] Issue in broadcastable converter utility #2227

Closed
@gs-olive

Description

@gs-olive

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.

Metadata

Metadata

Assignees

Labels

bugSomething isn't working

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions