Fix test flash_attn_varlen_func ut (#2578)

* Fix test flash_attn_varlen_func ut

* rm skipif

Author: 0x45f

tests/test_flash_attn_varlen_func.py

@@ -93,7 +93,6 @@ def ref_paged_attn(
     return torch.cat(outputs, dim=0)
 
 
-@pytest.mark.skipif(True, reason="Dead code on line 221 fails to pass.")
 @pytest.mark.skipif(vendor_name == "kunlunxin", reason="RESULT TODOFIX")
 @pytest.mark.skipif(vendor_name == "hygon", reason="RESULT TODOFIX")
 @pytest.mark.flash_attn_varlen_func
@@ -229,6 +228,10 @@ def test_flash_attn_varlen_func(
                     softmax_scale=scale,
                     causal=causal,
                     window_size=window_size,
+                    block_table=block_tables,
+                    softcap=soft_cap if soft_cap is not None else 0,
+                    alibi_slopes=alibi_slopes,
+                    fa_version=2,
                 )
 
         ref_output = ref_paged_attn(