fix precommit

Signed-off-by: Yongye Zhu <zyy1102000@gmail.com>
2026-05-06 03:14:49 +00:00
parent c001535038
commit 97cd2c41ad
1 changed files with 0 additions and 4 deletions
@@ -70,8 +70,6 @@ class TokenspeedMLAPrefillBackend(MLAPrefillBackend):
        qk_rope_head_dim: int,
        v_head_dim: int,
        vllm_config: "VllmConfig",
-        device: torch.device,
-        layer_names: list[str] | None = None,
    ) -> None:
        super().__init__(
            num_heads=num_heads,
@@ -81,8 +79,6 @@ class TokenspeedMLAPrefillBackend(MLAPrefillBackend):
            qk_rope_head_dim=qk_rope_head_dim,
            v_head_dim=v_head_dim,
            vllm_config=vllm_config,
-            device=device,
-            layer_names=layer_names,
        )

        # Pre-JIT the kernel for both BF16 and FP8 prefill shapes so the first