onnxruntime
Update benchmark_mha.py to compare with PyTorch SDPA
#21449

Merged

Update benchmark_mha.py to compare with PyTorch SDPA #21449

tianleiwu merged 6 commits into main from tlwu/benchmark_torch_mha

update benchmark_mha to compare with PyTorch SDPA

6209dcf5

tianleiwu requested a review from

kunal-vaishnavi 1 year ago

tianleiwu requested a review from

wangyems 1 year ago

tianleiwu requested a review from

yufenglee 1 year ago

github-advanced-security commented on 2024-07-23

add comment

e4caf29d

remove unused variable

f3f95f2e

tianleiwu marked this pull request as draft 1 year ago

Add Windows batch file

ecff18ba

minor change

85f9a139

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

wangyems commented on 2024-07-25

wangyems approved these changes on 2024-07-26

kunal-vaishnavi commented on 2024-07-26

Merge branch 'main' into tlwu/benchmark_torch_mha

8fc1dc37

tianleiwu merged 64819f6f into main 1 year ago

tianleiwu deleted the tlwu/benchmark_torch_mha branch 1 year ago

Reviewers

wangyems

kunal-vaishnavi

yufenglee

Assignees

No one assigned

Labels

None yet

Milestone

No milestone