[Flex Attention] Apply patch from pytorch#143553 instead of using fork#3945
Merged
whitneywhtsang merged 1 commit intomainfrom Apr 16, 2025
Merged
[Flex Attention] Apply patch from pytorch#143553 instead of using fork#3945whitneywhtsang merged 1 commit intomainfrom
whitneywhtsang merged 1 commit intomainfrom