onnxruntime
[ROCM] adjust test_flash_attn_rocm test tolerance
#21379

Merged

[ROCM] adjust test_flash_attn_rocm test tolerance #21379

tianleiwu merged 1 commit into main from tlwu/flash_attn_rocm_tolerance

adjust gqa flash attention test threshold for rocm

ec83548f

tianleiwu requested a review from

cloudhan 1 year ago

cloudhan approved these changes on 2024-07-17

tianleiwu merged 0f4c39ec into main 1 year ago

tianleiwu deleted the tlwu/flash_attn_rocm_tolerance branch 1 year ago

Reviewers

cloudhan

Assignees

No one assigned

Labels

None yet

Milestone

No milestone