xla
[ragged-paged-attn] Unify kv strided load to one.
#8929

Merged

[ragged-paged-attn] Unify kv strided load to one. #8929

yaochengji merged 3 commits into pytorch:master from bythew3i:ragged-attn-v2

[ragged-paged-attn] Unify kv strided load to one.

bd69fbfa

bythew3i force pushed from ad66996b to bd69fbfa 1 year ago

Merge branch 'master' into ragged-attn-v2

9ab9f318

yaochengji approved these changes on 2025-04-03

Update ragged_paged_attention_v2.py

e109142b

vanbasten23 approved these changes on 2025-04-03

yaochengji merged c044c691 into master 1 year ago

Reviewers

vanbasten23

yaochengji

Assignees

No one assigned

Labels

None yet

Milestone

No milestone