update triton_decode_attention.py

085c3454 · zhuwenwen · 8b5a09f6 · 085c3454
Commit 085c3454 authored Apr 09, 2025 by zhuwenwen
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

vllm/attention/ops/triton_decode_attention.py vllm/attention/ops/triton_decode_attention.py +1 -1

No files found.
--- a/vllm/attention/ops/triton_decode_attention.py
+++ b/vllm/attention/ops/triton_decode_attention.py
@@ -1602,7 +1602,7 @@ def decode_attention_fwd(
                    logit_cap=logit_cap,
                )
            elif best_config['kernel_kind'] == 'v2_tc':
-                decode_attention_v1(
+                decode_attention_v2(
                    q,
                    k_buffer,
                    v_buffer,