xla
Adjust allclose atol for the flash attention TPU test
#6889

Merged

Adjust allclose atol for the flash attention TPU test #6889

JackCaoG merged 2 commits into master from JackCaoG/flash_attention_test_precision

Adjust allclose atol for the flash attention TPU test

a673b2f5

JackCaoG requested a review from

alanwaketan 1 year ago

JackCaoG requested a review from

lsy323 1 year ago

alanwaketan commented on 2024-04-04

lint

d7896b1e

alanwaketan approved these changes on 2024-04-04

JackCaoG merged 73de972b into master 1 year ago

Reviewers

alanwaketan

lsy323

Assignees

No one assigned

Labels

None yet

Milestone

No milestone