flexattention: port flex attention overflow fixes to master-pytorch-AtomGit

Star895

Star895

已合并

flexattention: port flex attention overflow fixes to master #45274

a_knight创建于 8月25日从a_knight/pytorch： master 合入到 Ascend/pytorch： master

flexattention: port flex attention overflow fixes to master #45274

已合并

a_knight创建于 8月25日

从a_knight/pytorch master合入到Ascend/pytorch master