update test_flash_mla.py

tests/kernels/test_flashmla.py

@@ -33,7 +33,8 @@ FLASH_MLA_UNSUPPORTED_REASON = is_flashmla_supported()[1] \
 @pytest.mark.parametrize("dv", [512])
 @pytest.mark.parametrize("block_size", [64])
 @pytest.mark.parametrize("causal", [True])
-@pytest.mark.parametrize("varlen", [False, True])
+# @pytest.mark.parametrize("varlen", [False, True])
+@pytest.mark.parametrize("varlen", [True])
 @torch.inference_mode()
 def test_flash_mla(b, s_q, mean_sk, h_q, h_kv, d, dv, block_size, causal,
                    varlen):