onnxruntime
Extend memory efficient attention coverage in Attention/MHA cuda op
#15064
Merged

Extend memory efficient attention coverage in Attention/MHA cuda op #15064

gh-yewang merged 23 commits into main from wangye/cutlass_bias
gh-yewang
update to cutlass 3.0
d4cd2846
Merge branch 'main' of github.com:microsoft/onnxruntime into wangye/c…
07984e82
rpb parity is good
f8f0d8ca
ut
eb1f8184
fix batch issue and restrict seq_len
be29652b
gh-yewang gh-yewang requested a review from tianleiwu tianleiwu 3 years ago
update deps version
b7ab0a13
update
d6f4c450
Merge branch 'main' of github.com:microsoft/onnxruntime into wangye/c…
b8eafc07
patch
7423668b
fix commplex.hpp
71a74ea0
Revert "fix commplex.hpp"
48644cf3
kunal-vaishnavi Attempting pipeline fix using guards
cc1c7ddd
kunal-vaishnavi Adding patch command to cutlass cmake
bc96ff20
kunal-vaishnavi Combine cutlass patch files
6944277b
Merge branch 'main' into wangye/cutlass_bias
6f93ad60
kunal-vaishnavi Adding updated contrib ops doc
960b96f2
gh-yewang gh-yewang marked this pull request as ready for review 3 years ago
gh-yewang gh-yewang requested a review 3 years ago
gh-yewang gh-yewang requested a review 3 years ago
support batched rpb
7128682a
refine patch file
1872fb7a
Merge branch 'main' of github.com:microsoft/onnxruntime into wangye/c…
ea3a8553
tianleiwu
tianleiwu commented on 2023-03-21
tianleiwu
tianleiwu commented on 2023-03-21
review comments
42e72f39
gh-yewang gh-yewang requested a review from tianleiwu tianleiwu 3 years ago
fix comment
bfefe03a
gh-yewang gh-yewang marked this pull request as draft 3 years ago
update docs
aad4941f
Merge branch 'main' of github.com:microsoft/onnxruntime into wangye/c…
b8be2bd6
gh-yewang gh-yewang marked this pull request as ready for review 3 years ago
tianleiwu
tianleiwu approved these changes on 2023-03-23
jchen351
jchen351 approved these changes on 2023-03-23
pranavsharma
pranavsharma commented on 2023-03-23
gh-yewang
pranavsharma
pranavsharma approved these changes on 2023-03-23
gh-yewang gh-yewang merged 2ee822d4 into main 3 years ago
gh-yewang gh-yewang deleted the wangye/cutlass_bias branch 3 years ago

Login to write a write a comment.

Login via GitHub

Assignees
No one assigned
Labels
Milestone