PR #24953 Allow FP16 math in flash attention

Allow FP16 math in flash attention #24953

sushraja-msft merged 3 commits into main from user/sushraja/fp16_fa

Return back to fp16 fa

7e1d816b

sushraja-msft requested a review from

guschmue 1 year ago

sushraja-msft requested a review from

qjia7 1 year ago

guschmue added ep:WebGPU

fs-eire commented on 2025-06-04

Make the min_value precision dependent

63f422b2

github-actions commented on 2025-06-04

Update onnxruntime/contrib_ops/webgpu/bert/flash_attention.cc

18cd745a

qjia7 approved these changes on 2025-06-05

fs-eire approved these changes on 2025-06-05

guschmue approved these changes on 2025-06-05

sushraja-msft merged 1c577b71 into main 363 days ago

sushraja-msft deleted the user/sushraja/fp16_fa branch 363 days ago

Reviewers

guschmue

qjia7

fs-eire

github-actions

tianleiwu

Assignees

No one assigned

Labels

ep:WebGPU

Milestone

No milestone