avoid a redundant contiguous() call for the end of the sm100 cutlass mla decode

Signed-off-by: Alexander Matveev <[email protected]>

Author: alexm-redhat

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

@@ -210,9 +210,14 @@ def _sm100_cutlass_mla_decode(
             sm_scale,
             num_kv_splits,
         )
-        returned_lse = lse[:, :H].contiguous(
-        ) if self.need_to_return_lse_for_decode else lse
-        return out[:, :H].contiguous(), returned_lse
+
+        if H < MAX_HEADS:
+            # Extract the subsets of the outputs
+            returned_lse = lse[:, :H].contiguous(
+            ) if self.need_to_return_lse_for_decode else lse
+            out = out[:, :H]
+
+        return out, returned_lse
 
     def _sm100_forward_decode(
         self,