Skip to content

Fix indexer mask padding in Tokamax Flash Attention with Context Parallelism - #4947

Draft
SurbhiJainUSC wants to merge 2 commits into
mainfrom
ci_tests
Draft

Fix indexer mask padding in Tokamax Flash Attention with Context Parallelism#4947
SurbhiJainUSC wants to merge 2 commits into
mainfrom
ci_tests

Fix dynamic mask splash attention mask processing and indexer boolean…

f8c784b
Select commit
Loading
Failed to load commit list.
Google CLA / cla/google succeeded Aug 20, 2026 in 8s

✅ All contributors are covered under a CLA with Google

See https://cla.developers.google.com/ for more info about Google's Contributor License Agreement (CLA).

ℹ️ Googlers: Go here to view more details and manage scans for this pull request.

Details

The following contributors were found for this pull request:

f8c784b Author: @SurbhiJainUSC <sj****hi​@google.com>

(Only the first commit for a unique contributor is listed.)