SIGN IN SIGN UP

[PE] Skip test_sdpa_can_dispatch_on_flash for TimmWrapper-backed models (#48064)

* [PE] Skip test_sdpa_can_dispatch_on_flash for TimmWrapper-backed models

TimmWrapper declares _supports_sdpa=True (timm uses F.scaled_dot_product_attention
internally), but timm does not expose an API to restrict to flash-only dispatch.
The test forces sdpa_kernel(enable_flash=True, enable_math=False, enable_mem_efficient=False),
which causes RuntimeError: No available kernel for PeAudioVideoEncoderTest and
PeVideoEncoderTest. Skip until timm supports controllable attention dispatch.

Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>

* Remove extra skip from PeVideoModelTest (only EncoderTest reported failing in CI)

Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>

* Fix indentation error left by previous edit

Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>

---------

Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
Y
Yih-Dar committed
f9b76f2dfc44fdc6109468925bf0e1856fd34278
Parent: 80c6674
Committed by GitHub <noreply@github.com> on 8/18/2026, 4:09:14 PM