SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

[tests] make test_sdpa_equivalence device-agnostic (#32520)

Browse Source 
* fix on xpu

* [run_all]
This commit is contained in:
Fanli Lin
2024-08-16 18:34:13 +08:00
committed by GitHub
parent 70d5df6107
commit 8f9fa3b081
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
tests/models/gemma/test_modeling_gemma.py
View File

@@ -27,6 +27,7 @@ from transformers.testing_utils import (
require_flash_attn, require_flash_attn,
require_read_token, require_read_token,
require_torch, require_torch,
require_torch_accelerator,
require_torch_gpu, require_torch_gpu,
require_torch_sdpa, require_torch_sdpa,
slow, slow,
@@ -460,7 +461,7 @@ class GemmaModelTest(ModelTesterMixin, GenerationTesterMixin, PipelineTesterMixi
self.skipTest(reason="Gemma flash attention does not support right padding") self.skipTest(reason="Gemma flash attention does not support right padding")
@require_torch_sdpa @require_torch_sdpa
@require_torch_gpu @require_torch_accelerator
@slow @slow
def test_sdpa_equivalence(self): def test_sdpa_equivalence(self):
for model_class in self.all_model_classes: for model_class in self.all_model_classes: