xla
flash_attention: support also cross attention.
#8427

Merged

flash_attention: support also cross attention. #8427

JackCaoG merged 1 commit into pytorch:master from dudulightricks:bug-fix/support-cross-flash-attn

JackCaoG added tpuci

dudulightricks force pushed from 9716dc06 to 29be2c65 1 year ago

dudulightricks force pushed from 29be2c65 to 840e3fe7 1 year ago

dudulightricks force pushed from 840e3fe7 to 1bc1fb4d 1 year ago

flash_attention: support also cross attention.

9e1e24be

dudulightricks force pushed from 1bc1fb4d to 9e1e24be 1 year ago

JackCaoG approved these changes on 2024-12-10

JackCaoG merged 5790092f into master 1 year ago

Reviewers

JackCaoG

Assignees

No one assigned

Labels

None yet

Milestone

No milestone