Skip to content

[GPU] Fix accuracy issues for sdpa_micro #30698

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

Merged
merged 4 commits into from
May 27, 2025

Conversation

ahnyoung-paul
Copy link
Contributor

@ahnyoung-paul ahnyoung-paul commented May 23, 2025

Details:

  • Fix attn mask tile loading issue when attn mask shape has single query dimension [batch, num_heads, 1, sequence_length of k]

Tickets:

  • 166691

@ahnyoung-paul ahnyoung-paul requested review from a team as code owners May 23, 2025 13:52
@ahnyoung-paul ahnyoung-paul added category: GPU OpenVINO GPU plugin pr: needs tests PR needs tests updating under_perf_check labels May 23, 2025
if (MSK_D2 == 1) {
// Define mask dimensions for single Query dimension
uint mask_m = MSK_D1; // num_heads
uint mask_n = MSK_D3; // sequence_length
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

why num_heads? masks are usually [q, seqlen]..

Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Yes, you're right. I fixed codes. please review it again. thanks.

@ahnyoung-paul ahnyoung-paul changed the title [GPU] Fix accuracy issues for mvn and sdpa_micro [GPU] Fix accuracy issues for sdpa_micro May 26, 2025
@sshlyapn
Copy link
Contributor

@ahnyoung-paul please add some test case for this scenario

@ahnyoung-paul
Copy link
Contributor Author

@ahnyoung-paul please add some test case for this scenario

I'm making unit test for this case. I will update it soon.

@ahnyoung-paul ahnyoung-paul removed the pr: needs tests PR needs tests updating label May 26, 2025
@ahnyoung-paul ahnyoung-paul requested review from sshlyapn and yeonbok May 27, 2025 02:02
@yeonbok yeonbok added this pull request to the merge queue May 27, 2025
Merged via the queue into openvinotoolkit:master with commit 3b84486 May 27, 2025
182 checks passed
@yeonbok yeonbok deleted the fix_ltx_video_acc branch May 27, 2025 22:26
ahnyoung-paul added a commit to ahnyoung-paul/openvino that referenced this pull request May 28, 2025
### Details:
- *Fix attn mask tile loading issue when attn mask shape has single
query dimension [batch, num_heads, 1, sequence_length of k]*

### Tickets:
 - *166691*
github-merge-queue bot pushed a commit that referenced this pull request May 28, 2025
### Details:
- *Fix attn mask tile loading issue when attn mask shape has single
query dimension [batch, num_heads, 1, sequence_length of k]*

### Tickets:
 - *166691*
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants