SIGN IN SIGN UP

Support flashinfer for Gemma3 prefill (#3167)

* launcher: ensure correct detection of Gemma 3 head size

* Support flashinfer for Gemma3 prefill

Gemma3 uses bidirectional attention for images. Flashinfer
supports custom masks. Hook up the mask with flashinfer, so that we do
not have to use the slower SDPA implementation for prefills with images.

* Update Gemma3 test outputs

* Fixed unused import
D
Daniël de Kok committed
84ab88d84343fabef2baa89885e9efb55907f75e
Parent: 4645678
Committed by GitHub <noreply@github.com> on 4/17/2025, 4:07:41 PM