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

[GPU] Fix incorrect broadcast axis for static SDPA case #28763

Open
wants to merge 1 commit into
base: master
Choose a base branch
from
Open

[GPU] Fix incorrect broadcast axis for static SDPA case #28763

wants to merge 1 commit into from
+48 −18

Conversation

sshlyapn
Copy link
Contributor

Details:

  • This change fixes incorrect broadcast axis calculation for static SDPA operations (dynamic case wasn't affected because the only static dimensions were head_size and heads_num, which allowed the proper setting of broadcast axis)

Tickets:

@sshlyapn sshlyapn added the category: GPU OpenVINO GPU plugin label Jan 31, 2025
@sshlyapn sshlyapn added this to the 2025.1 milestone Jan 31, 2025
@sshlyapn sshlyapn requested review from a team as code owners January 31, 2025 12:50
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Reviewers
No reviews
Assignees
No one assigned
Labels
category: GPU OpenVINO GPU plugin
Projects
None yet
Milestone
2025.1
Development

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