[MoE] Fix output_shape calculation in Attention layer to handle 3D query inputs#31596

LucasWilkinson merged 1 commit intovllm-project:mainvllm-project/vllm:mainfrom

ROCm:akaratza_fix_moe_out_shapeROCm/vllm:akaratza_fix_moe_out_shapeCopy head branch name to clipboard