PR #8771 Add sm_scale in ragged attention kernel

Add sm_scale in ragged attention kernel #8771

yaochengji merged 4 commits into master from chengji/fix-sm-scale

fix sm_scale

5c690075

add tests

1120fcde

yaochengji requested a review from

vanbasten23 359 days ago

vanbasten23 commented on 2025-02-28

vanbasten23 approved these changes on 2025-02-28

fix lint

04ceed5a

remove unnessary astype

e7bd5f2d

yaochengji merged 2675e689 into master 359 days ago

Reviewers

vanbasten23

Assignees

No one assigned

Labels

None yet

Milestone

No milestone