### Feature request With the recent publish of `FlexAttention`, think it makes senses to support it with sdpa path of `Gemma2`. Leaving it open to the community: - [ ] implement the path - [ ] make sure integration tests work - [ ] add some doc Related to #32865
Feature request
With the recent publish of
FlexAttention, think it makes senses to support it with sdpa path ofGemma2.Leaving it open to the community:
Related to #32865