xla
fix ab in flash attention
#8540
Merged

fix ab in flash attention #8540

ZhiyuLi-goog
ZhiyuLi-goog fix ab in flash attention
06bc53bc
ZhiyuLi-goog
tengyifei
tengyifei approved these changes on 2025-01-07
tengyifei tengyifei added tpuci
qihqi qihqi merged c673809a into master 1 year ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
Milestone