Compare commits

...

2 Commits

9 changed files with 93 additions and 0 deletions

View File

@ -502,6 +502,14 @@ class Blip2ForConditionalGenerationDecoderOnlyTest(ModelTesterMixin, GenerationT
def test_model_get_set_embeddings(self):
pass
@unittest.skip(reason="See issue #38820")
def test_sdpa_can_compile_dynamic(self):
pass
@unittest.skip(reason="See issue #38820")
def test_sdpa_can_dispatch_on_flash(self):
pass
@require_torch_sdpa
def test_sdpa_can_dispatch_composite_models(self):
"""
@ -935,6 +943,14 @@ class Blip2ModelTest(ModelTesterMixin, PipelineTesterMixin, GenerationTesterMixi
def test_cpu_offload(self):
pass
@unittest.skip(reason="See issue #38820")
def test_sdpa_can_compile_dynamic(self):
pass
@unittest.skip(reason="See issue #38820")
def test_sdpa_can_dispatch_on_flash(self):
pass
@require_torch_sdpa
def test_sdpa_can_dispatch_composite_models(self):
"""
@ -1203,6 +1219,14 @@ class Blip2TextModelWithProjectionTest(ModelTesterMixin, unittest.TestCase):
def test_model_common_attributes(self):
pass
@unittest.skip(reason="See issue #38820")
def test_sdpa_can_compile_dynamic(self):
pass
@unittest.skip(reason="See issue #38820")
def test_sdpa_can_dispatch_on_flash(self):
pass
def test_forward_signature(self):
config, _ = self.model_tester.prepare_config_and_inputs_for_common()
@ -1361,6 +1385,14 @@ class Blip2VisionModelWithProjectionTest(ModelTesterMixin, unittest.TestCase):
def test_retain_grad_hidden_states_attentions(self):
pass
@unittest.skip(reason="See issue #38820")
def test_sdpa_can_compile_dynamic(self):
pass
@unittest.skip(reason="See issue #38820")
def test_sdpa_can_dispatch_on_flash(self):
pass
def test_model_common_attributes(self):
config, _ = self.model_tester.prepare_config_and_inputs_for_common()
@ -1504,6 +1536,14 @@ class Blip2TextRetrievalModelTest(ModelTesterMixin, unittest.TestCase):
def test_model_common_attributes(self):
pass
@unittest.skip(reason="See issue #38820")
def test_sdpa_can_compile_dynamic(self):
pass
@unittest.skip(reason="See issue #38820")
def test_sdpa_can_dispatch_on_flash(self):
pass
def test_forward_signature(self):
config, _ = self.model_tester.prepare_config_and_inputs_for_common()

View File

@ -318,6 +318,14 @@ class DeepseekV3ModelTest(ModelTesterMixin, GenerationTesterMixin, PipelineTeste
def test_greedy_generate_dict_outputs_use_cache(self):
pass
@unittest.skip(reason="See issue #38820")
def test_sdpa_can_compile_dynamic(self):
pass
@unittest.skip(reason="See issue #38820")
def test_sdpa_can_dispatch_on_flash(self):
pass
def test_config(self):
self.config_tester.run_common_tests()

View File

@ -194,6 +194,10 @@ class DPRModelTest(ModelTesterMixin, PipelineTesterMixin, unittest.TestCase):
self.model_tester = DPRModelTester(self)
self.config_tester = ConfigTester(self, config_class=DPRConfig, hidden_size=37)
@unittest.skip(reason="See issue #38820")
def test_sdpa_can_dispatch_on_flash(self):
pass
def test_config(self):
self.config_tester.run_common_tests()

View File

@ -198,6 +198,14 @@ class DPTModelTest(ModelTesterMixin, PipelineTesterMixin, unittest.TestCase):
def test_inputs_embeds(self):
pass
@unittest.skip(reason="See issue #38820")
def test_sdpa_can_compile_dynamic(self):
pass
@unittest.skip(reason="See issue #38820")
def test_sdpa_can_dispatch_on_flash(self):
pass
def test_model_get_set_embeddings(self):
config, _ = self.model_tester.prepare_config_and_inputs_for_common()

View File

@ -352,6 +352,10 @@ class Gemma3Vision2TextModelTest(ModelTesterMixin, GenerationTesterMixin, unitte
def test_initialization(self):
pass
@unittest.skip(reason="See issue #38820")
def test_sdpa_can_dispatch_on_flash(self):
pass
def test_automodelforcausallm(self):
"""
Regression test for #36741/#36917 -- make sure `AutoModelForCausalLM` works with a Gemma3 config, i.e. that

View File

@ -447,6 +447,10 @@ class GPTBigCodeModelTest(ModelTesterMixin, GenerationTesterMixin, PipelineTeste
def test_generate_continue_from_inputs_embeds(self):
pass
@unittest.skip(reason="See issue #38820")
def test_sdpa_can_dispatch_on_flash(self):
pass
def test_gpt_bigcode_model(self):
config_and_inputs = self.model_tester.prepare_config_and_inputs()
self.model_tester.create_and_check_gpt_bigcode_model(*config_and_inputs)

View File

@ -307,6 +307,15 @@ class GraniteSpeechForConditionalGenerationModelTest(ModelTesterMixin, Generatio
if "SdpaAttention" in class_name or "SdpaSelfAttention" in class_name:
raise ValueError("The eager model should not have SDPA attention layers")
@unittest.skip(reason="See issue #38820")
def test_sdpa_can_compile_dynamic(self):
pass
@unittest.skip(reason="See issue #38820")
def test_sdpa_can_dispatch_on_flash(self):
pass
@pytest.mark.generate
@require_torch_sdpa
@slow

View File

@ -526,6 +526,14 @@ class InstructBlipForConditionalGenerationDecoderOnlyTest(ModelTesterMixin, Gene
def test_generate_from_inputs_embeds_with_static_cache(self):
pass
@unittest.skip(reason="See issue #38820")
def test_sdpa_can_compile_dynamic(self):
pass
@unittest.skip(reason="See issue #38820")
def test_sdpa_can_dispatch_on_flash(self):
pass
def test_forward_signature(self):
config, _ = self.model_tester.prepare_config_and_inputs_for_common()

View File

@ -541,6 +541,14 @@ class InstructBlipVideoForConditionalGenerationDecoderOnlyTest(
def test_generate_from_inputs_embeds_with_static_cache(self):
pass
@unittest.skip(reason="See issue #38820")
def test_sdpa_can_compile_dynamic(self):
pass
@unittest.skip(reason="See issue #38820")
def test_sdpa_can_dispatch_on_flash(self):
pass
def test_forward_signature(self):
config, _ = self.model_tester.prepare_config_and_inputs_for_common()