flex_attention ver. #192

joydddd wants to merge 2 commits intopytorch-labs:mainpytorch-labs/gpt-fast:mainfrom

joydddd:flex_attjoydddd/gpt-fast:flex_attCopy head branch name to clipboard