[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