Skip to content
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

add scalarization patterns to support dynamic pytorch pad exports #3838

Merged
merged 13 commits into from
Nov 1, 2024

Conversation

zjgarvey
Copy link
Collaborator

@zjgarvey zjgarvey commented Oct 30, 2024

  1. Adds case handling for aten.slice.tensor shape inference with negative strides. This is not technically allowed by native pytorch, but it is useful for ONNX ingest. We were getting some incorrect shapes for these negative strided slice ops.
  2. Adds scalarization support for ops seen in pytorch pad exports to ONNX. These are typically aten.view aten.transpose.int and aten.slice.Tensor with negative strides (and rank 2).
  3. Allows view op self to be added to the worklist conditionally, based on whether the view op actually occurs as a middle point in a shape computation.

@zjgarvey zjgarvey merged commit 738d45d into llvm:main Nov 1, 2024
3 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants