[tests] Parameterized test_eager_matches_sdpa_inference (#36650)

2025-03-14 14:41:27 +00:00
parent 9215cc62d4
commit 42ebb6c23e
16 changed files with 285 additions and 1900 deletions
--- a/tests/models/recurrent_gemma/test_modeling_recurrent_gemma.py
+++ b/tests/models/recurrent_gemma/test_modeling_recurrent_gemma.py
@@ -347,10 +347,6 @@ class RecurrentGemmaModelTest(ModelTesterMixin, PipelineTesterMixin, unittest.Te
    def test_eager_matches_sdpa_generate(self):
        pass

-    @unittest.skip(reason="RecurrentGemma only supports sdpa")
-    def test_eager_matches_sdpa_inference(self):
-        pass
-
    @unittest.skip(reason="RecurrentGemma does not return the cache")
    def test_contrastive_generate_low_memory(self):
        pass