Repository: huggingface/transformers
Pull request: skip test_sdpa_can_dispatch_on_flash in tests/models/pe_audio/test_mo…

…deling_pe_audio.py::PeAudioEncoderTest

```
tests/models/pe_audio/test_modeling_pe_audio.py::PeAudioEncoderTest::test_sdpa_can_dispatch_on_flash
  /mnt/disk3/wangyi/transformers/src/transformers/integrations/sdpa_attention.py:92: UserWarning: Memory efficient kernel not used because: (Trigger$d internally at /pytorch/aten/src/ATen/native/transformers/cuda/sdp_utils.cpp:915.)
    attn_output = torch.nn.functional.scaled_dot_product_attention(

tests/models/pe_audio/test_modeling_pe_audio.py::PeAudioEncoderTest::test_sdpa_can_dispatch_on_flash
  /mnt/disk3/wangyi/transformers/src/transformers/integrations/sdpa_attention.py:92: UserWarning: Memory Efficient attention has been runtime disabl$d. (Triggered internally at /pytorch/aten/src/ATen/native/transformers/sdp_utils_cpp.h:552.)
    attn_output = torch.nn.functional.scaled_dot_product_attention(

tests/models/pe_audio/test_modeling_pe_audio.py::PeAudioEncoderTest::test_sdpa_can_dispatch_on_flash
  /mnt/disk3/wangyi/transformers/src/transformers/integrations/sdpa_attention.py:92: UserWarning: Flash attention kernel not used because: (Triggere$ internally at /pytorch/aten/src/ATen/native/transformers/cuda/sdp_utils.cpp:917.)
    attn_output = torch.nn.functional.scaled_dot_product_attention(

tests/models/pe_audio/test_modeling_pe_audio.py::PeAudioEncoderTest::test_sdpa_can_dispatch_on_flash
  /mnt/disk3/wangyi/transformers/src/transformers/integrations/sdpa_attention.py:92: UserWarning: Flash Attention does not support non-null attn_mas$. (Triggered internally at /pytorch/aten/src/ATen/native/transformers/sdp_utils_cpp.h:262.)
    attn_output = torch.nn.functional.scaled_dot_product_attention(

tests/models/pe_audio/test_modeling_pe_audio.py::PeAudioEncoderTest::test_sdpa_can_dispatch_on_flash
  /mnt/disk3/wangyi/transformers/src/transformers/integrations/sdpa_attention.py:92: UserWarning: cuDNN attention kernel not used because: (Triggere$ internally at /pytorch/aten/src/ATen/native/transformers/cuda/sdp_utils.cpp:919.)
    attn_output = torch.nn.functional.scaled_dot_product_attention(

tests/models/pe_audio/test_modeling_pe_audio.py::PeAudioEncoderTest::test_sdpa_can_dispatch_on_flash
  /mnt/disk3/wangyi/transformers/src/transformers/integrations/sdpa_attention.py:92: UserWarning: cuDNN attention has been runtime disabled. (Trigge$ed internally at /pytorch/aten/src/ATen/native/transformers/cuda/sdp_utils.cpp:635.)
    attn_output = torch.nn.functional.scaled_dot_product_attention(

-- Docs: https://docs.pytest.org/en/stable/how-to/capture-warnings.html
============================================================== short test summary info =============================================================$
FAILED tests/models/pe_audio/test_modeling_pe_audio.py::PeAudioEncoderTest::test_sdpa_can_dispatch_on_flash - RuntimeError: No available kernel. Aborting execution.

```
@ydshieh please help review, thanks


Changed files:
filename: tests/models/pe_audio/test_modeling_pe_audio.py
patch: @@ -158,6 +158,10 @@ def test_model_get_set_embeddings(self):
     def test_feed_forward_chunking(self):
         pass
 
+    @unittest.skip(reason="SDPA can't dispatch on flash with not None `attention_mask`")
+    def test_sdpa_can_dispatch_on_flash(self):
+        pass
+
 
 class PeAudioTextModelTester:
     """