[FlashInfer] Upgrade to 0.2.0 (#11194)

Signed-off-by: Bowen Wang <abmfy@icloud.com> Signed-off-by: youkaichao <youkaichao@gmail.com> Co-authored-by: youkaichao <youkaichao@gmail.com>
2025-01-28 02:19:24 +08:00
parent 3f1fc7425a
commit 2bc3fbba0c
10 changed files with 256 additions and 74 deletions
--- a/tests/compile/test_basic_correctness.py
+++ b/tests/compile/test_basic_correctness.py
@@ -58,7 +58,7 @@ test_settings = [
        model_args=["--task", "embed"],
        pp_size=1,
        tp_size=1,
-        attn_backend="FLASHINFER",
+        attn_backend="FLASH_ATTN",
        method="encode",
        fullgraph=True,
    ),