refactor: keep problem shape in fmha block for q/k/v shape#495

guocuimi merged 1 commit intomainvectorch-ai/ScaleLLM:mainfrom

attn_problem_shapevectorch-ai/ScaleLLM:attn_problem_shapeCopy head branch name to clipboard