Skip to content

🐛 [Bug] Issue in broadcastable converter utility #2227

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Closed
gs-olive opened this issue Aug 14, 2023 · 0 comments · Fixed by #2228
Closed

🐛 [Bug] Issue in broadcastable converter utility #2227

gs-olive opened this issue Aug 14, 2023 · 0 comments · Fixed by #2228
Assignees
Labels
bug Something isn't working

Comments

@gs-olive
Copy link
Collaborator

gs-olive commented Aug 14, 2023

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.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
bug Something isn't working
Projects
None yet
Development

Successfully merging a pull request may close this issue.

1 participant