Skip to content

[webgpu] Fix multihead-attention for ort-web-tests#24485

Merged
fs-eire merged 2 commits intomicrosoft:mainfrom
jchen10:mha
Apr 29, 2025
Merged

[webgpu] Fix multihead-attention for ort-web-tests#24485
fs-eire merged 2 commits intomicrosoft:mainfrom
jchen10:mha

Conversation

@jchen10
Copy link
Contributor

@jchen10 jchen10 commented Apr 21, 2025

Compute 'total_sequence_length' the same way as JSEP.

@jchen10
Copy link
Contributor Author

jchen10 commented Apr 21, 2025

The corresponding JSEP code line.
@xhcao @fs-eire PTAL

@guschmue guschmue added the ep:WebGPU ort-web webgpu provider label Apr 24, 2025
@guschmue
Copy link
Contributor

/azp run Linux QNN CI Pipeline,Win_TRT_Minimal_CUDA_Test_CI,Windows ARM64 QNN CI Pipeline,Windows GPU Doc Gen CI Pipeline,Windows x64 QNN CI Pipeline

@azure-pipelines
Copy link

Azure Pipelines successfully started running 5 pipeline(s).

@fs-eire
Copy link
Contributor

fs-eire commented Apr 29, 2025

some pipeline failed, maybe need to merge to latest main

@jchen10
Copy link
Contributor Author

jchen10 commented Apr 29, 2025

some pipeline failed, maybe need to merge to latest main

Thanks for the reminder. Rebased to main!

@fs-eire
Copy link
Contributor

fs-eire commented Apr 29, 2025

/azp run Linux QNN CI Pipeline,Win_TRT_Minimal_CUDA_Test_CI,Windows ARM64 QNN CI Pipeline,Windows GPU Doc Gen CI Pipeline,Windows x64 QNN CI Pipeline

@azure-pipelines
Copy link

Azure Pipelines successfully started running 5 pipeline(s).

@fs-eire fs-eire merged commit 4513476 into microsoft:main Apr 29, 2025
70 checks passed
ankitm3k pushed a commit to intel/onnxruntime that referenced this pull request May 12, 2025
Compute 'total_sequence_length' the same way as JSEP.
@jchen10 jchen10 deleted the mha branch September 2, 2025 11:38
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

ep:WebGPU ort-web webgpu provider

Projects

None yet

Development

Successfully merging this pull request may close these issues.

5 participants