PR #8791 Integrate ragged paged attention v2

Integrate ragged paged attention v2 #8791

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

Add ragged paged attn v2

aa8c0261

Update custom kernel

4064de16

Update tests

6ab903c0

yaochengji requested a review from

yaochengji 245 days ago

yaochengji approved these changes on 2025-03-05

yaochengji enabled auto-merge (squash) 245 days ago

vanbasten23 approved these changes on 2025-03-05

yaochengji merged 5644f44a into master 245 days ago

Reviewers

vanbasten23

yaochengji

Assignees

No one assigned

Labels

None yet

Milestone

No milestone