update flashmla.py

vllm/v1/attention/backends/mla/flashmla.py

@@ -167,7 +167,7 @@ class FlashMLAImpl(MLACommonImpl[FlashMLAMetadata]):
         assert attn_metadata.decode is not None
 
         if envs.VLLM_USE_TRITON_CAT:
-            q = concat_helper([q_nope, q_pe], dim=-1)\
+            q = concat_helper(q_nope, q_pe, dim=-1)\
                 .unsqueeze(1)
         else:
             q = torch.cat([q_nope, q_pe], dim=-1)\