PR #21028 refactor flash attn test

refactor flash attn test #21028

tianleiwu merged 5 commits into main from guangyunhan/refactor-flash-attn-test-2

test: refactor flash_attn cuda test to use numpy.testing.assert_allclose

2b3f9e80

test: extend flash_attn test to run with non cuda ep

e8126feb

cloudhan requested a review from

aciddelgado 1 year ago

cloudhan requested a review from

tianleiwu 1 year ago

tianleiwu commented on 2024-06-13

github-advanced-security commented on 2024-06-13

test: show config with err_msg

22c92cd4

cloudhan force pushed from f529ef56 to 22c92cd4 1 year ago

lint

1002caf0

tianleiwu commented on 2024-06-13

test: make err_msg more verbose

d8902938

tianleiwu approved these changes on 2024-06-14

tianleiwu merged f4b22f89 into main 1 year ago

tianleiwu deleted the guangyunhan/refactor-flash-attn-test-2 branch 1 year ago

Reviewers

tianleiwu

aciddelgado

Assignees

No one assigned

Labels

None yet

Milestone

No milestone