[release/0.4][New features] Support FlashMask with 512/512 and 576/512 head dimensions - #1736

[release/0.4][New features] Support FlashMask with 512/512 and 576/512 head dimensions#1736

risemeup1111 wants to merge 1 commit into

PaddlePaddle:release/0.4PaddlePaddle/PaddleFleet:release/0.4from

risemeup1111:cherry-pick/1724/release/0.4risemeup1111/PaddleFleet:cherry-pick/1724/release/0.4Copy head branch name to clipboard