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.

Sign up for GitHub

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

Jump to bottom

Extend Einsum Core and common transformation to support broadcasting, repeated labels and ellipsis #28151

Open
wants to merge 19 commits into
base: master
Choose a base branch
from
Open

Extend Einsum Core and common transformation to support broadcasting, repeated labels and ellipsis #28151

wants to merge 19 commits into from
+1,084 −145

Conversation

mmikolajcz
Copy link
Contributor

@mmikolajcz mmikolajcz commented Dec 19, 2024 •
edited
Loading

Details:

  • Extend core implementation to support extended broadcasting to comply with PyTorch op
  • Extend common decomposition to support broadcasting of same label, ellipsis and repeated labels
  • Repeated labels subgraph that doesn't depend on static shapes,
  • Support of ellipsis label that doesn't mark any dimension,
  • *Test cases"

Tickets:

@github-actions github-actions bot added category: Core OpenVINO Core (aka ngraph) category: transformations OpenVINO Runtime library - Transformations labels Dec 19, 2024
@github-actions github-actions bot added the category: ONNX FE OpenVINO ONNX FrontEnd label Jan 8, 2025
@mmikolajcz
Remove Einsum xfail for torch HF tests
6796536 
Signed-off-by: MATEUSZ MIKOLAJCZYK <[email protected]>
@github-actions github-actions bot added the category: PyTorch FE OpenVINO PyTorch Frontend label Jan 8, 2025
@github-actions github-actions bot added the category: TEMPLATE OpenVINO Template plugin label Jan 21, 2025
@mmikolajcz mmikolajcz changed the title [WIP] Extend Einsum Core and common transformation to support broadcasting, repeated labels and ellipsis Extend Einsum Core and common transformation to support broadcasting, repeated labels and ellipsis Jan 23, 2025
@mmikolajcz mmikolajcz marked this pull request as ready for review January 23, 2025 18:09
@mmikolajcz mmikolajcz requested review from a team as code owners January 23, 2025 18:09
@mmikolajcz mmikolajcz requested review from a team as code owners January 23, 2025 18:09
@mmikolajcz mmikolajcz requested review from mvafin and itikhono and removed request for a team January 23, 2025 18:09
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Reviewers

@mitruska mitruska Awaiting requested review from mitruska

@rkazants rkazants Awaiting requested review from rkazants

@mvafin mvafin Awaiting requested review from mvafin mvafin is a code owner automatically assigned from openvinotoolkit/openvino-pytorch-frontend-maintainers

@itikhono itikhono Awaiting requested review from itikhono itikhono is a code owner automatically assigned from openvinotoolkit/openvino-ie-transformations-maintainers

At least 1 approving review is required to merge this pull request.

Assignees
No one assigned
Labels
category: Core OpenVINO Core (aka ngraph) category: ONNX FE OpenVINO ONNX FrontEnd category: PyTorch FE OpenVINO PyTorch Frontend category: TEMPLATE OpenVINO Template plugin category: transformations OpenVINO Runtime library - Transformations
Projects
None yet
Milestone
No milestone
Development

Successfully merging this pull request may close these issues.
1 participant
@mmikolajcz