[AMD] Fix Qwen3-Coder-Next: Add missing k_scale/v_scale args to extend_attention_fwd in aiter_backend (#19736)
This commit is contained in:
@@ -1765,6 +1765,8 @@ class AiterAttnBackend(AttentionBackend):
|
|||||||
True, # causal
|
True, # causal
|
||||||
self.forward_metadata.mask_indptr,
|
self.forward_metadata.mask_indptr,
|
||||||
self.forward_metadata.max_extend_len,
|
self.forward_metadata.max_extend_len,
|
||||||
|
1.0, # k_scale
|
||||||
|
1.0, # v_scale
|
||||||
layer.scaling,
|
layer.scaling,
|
||||||
logit_cap=layer.logit_cap,
|
logit_cap=layer.logit_cap,
|
||||||
)
|
)
|
||||||
|
|||||||
Reference in New Issue
Block a user