fix(flash_mla): fix a shape bug in flash_mla test#135

caizhi-mt wants to merge 1 commit into

bytedance:mainbytedance/xpu-perf:mainfrom

caizhi-mt:fix_flash_mla_testcaizhi-mt/ByteMLPerf:fix_flash_mla_testCopy head branch name to clipboard