transformers
[tests] Some fixes for test suite failures on TPU
#49425
Merged

[tests] Some fixes for test suite failures on TPU #49425

ydshieh merged 13 commits into main from tpu-test-fixes
tengomucho
tengomucho test: use the CUDA SDPA tolerances on TPU
48c16368
tengomucho test: request full-precision float32 matmuls in the test suite
ac290f35
tengomucho test: run subprocess tests in-process on single-process devices
1413987c
tengomucho test: restore the default device when the global device test fails
2fdb832f
tengomucho test: add TPU expectations to the gemma3 integration tests
39d05951
tengomucho test: add TPU expectations to the gemma3n integration tests
9f4fe385
tengomucho test: add TPU expectations to the gpt2 integration tests
78d4b41f
tengomucho test: add TPU expectations to the qwen2 integration tests
0a34359d
tengomucho test: add TPU expectations to the llama integration tests
9fc783e1
tengomucho test: add TPU expectations to the internvl integration tests
91bd6366
tengomucho test: add TPU expectations to the mistral3 integration tests
953b6117
tengomucho tengomucho requested a review from tarekziade tarekziade 26 days ago
tengomucho tengomucho requested a review from ydshieh ydshieh 26 days ago
tengomucho tengomucho requested a review from ArthurZucker ArthurZucker 26 days ago
tengomucho tengomucho requested a review from Cyrilvallez Cyrilvallez 26 days ago
tengomucho tengomucho requested a review from vasqu vasqu 26 days ago
github-actions
tengomucho
HuggingFaceDocBuilderDev
vasqu
vasqu commented on 2026-10-08
tengomucho test: add TPU expectations to the whisper integration tests
2d5b3156
github-actions
ydshieh
ydshieh commented on 2026-10-09
ydshieh
ydshieh approved these changes on 2026-10-09
ydshieh ydshieh changed the title [tests] Some fixes for test suite failures on TPU [DON'T MERGE YET] [tests] Some fixes for test suite failures on TPU 25 days ago
tengomucho review(test): relax whisper expectations on TPU
7aaed04c
github-actions
ydshieh ydshieh changed the title [DON'T MERGE YET] [tests] Some fixes for test suite failures on TPU [tests] Some fixes for test suite failures on TPU 23 days ago
ydshieh ydshieh merged 55c3cc9f into main 23 days ago
ydshieh ydshieh deleted the tpu-test-fixes branch 23 days ago

Login to write a write a comment.

Login via GitHub

Assignees
No one assigned
Labels
Milestone