onnxruntime
[CUDA] Update benchmark_mha.py to capture debug info to identify sdpa kernel
#21804
Merged

[CUDA] Update benchmark_mha.py to capture debug info to identify sdpa kernel #21804

tianleiwu
tianleiwu use debug info to identify sdpa_kernel in benchmark_mha
c94f18df
tianleiwu tianleiwu requested a review from yufenglee yufenglee 1 year ago
tianleiwu tianleiwu requested a review from gh-yewang gh-yewang 1 year ago
tianleiwu tianleiwu requested a review from kunal-vaishnavi kunal-vaishnavi 1 year ago
github-advanced-security
github-advanced-security commented on 2024-08-20
tianleiwu fix warnings that captured may be used before it is initialized
40834165
github-advanced-security
github-advanced-security commented on 2024-08-20
tianleiwu initialize captured_text
35410c7a
gh-yewang
gh-yewang approved these changes on 2024-08-22
tianleiwu tianleiwu merged 25d7a4fa into main 1 year ago
tianleiwu tianleiwu deleted the tlwu/benchmark_mha_kernel_from_debug_info branch 1 year ago

Login to write a write a comment.

Login via GitHub

Assignees
No one assigned
Labels
Milestone