PR #8741 Add padding ragged paged attention test

Add padding ragged paged attention test #8741

vanbasten23 merged 9 commits into master from xiowei/add_padding_ragged_paged_attention_test

add more tests about padding.

310fb810

changed the torch ref impl and add the dynamo test for the padded case.

d4b06cec

fix test that are hbm oom

abb6b76f

add the test for dynamo wrapper

d5864912

add the test for dynamo wrapper

44815b27

a not working version. show to Chengji. RUn python pytorch/xla/test/t…

7a9809b4

fix the test finally.

936f372f

linter

f43c6d23

remove pdb

4693a3e1

vanbasten23 marked this pull request as ready for review 1 year ago

vanbasten23 requested a review from

yaochengji 1 year ago

yaochengji approved these changes on 2025-02-25

vanbasten23 merged 5e1478d6 into master 1 year ago

miladm added pallas

Reviewers

yaochengji

Assignees

No one assigned

Labels

pallas

Milestone

No milestone