From d6e94176528b7b1d7e24e2cfee0b9cc663b8769d Mon Sep 17 00:00:00 2001 From: ApsarasX Date: Tue, 6 May 2025 21:54:52 +0800 Subject: [PATCH] [Bugfix] Fix masked_fill_ function typo (#769) ### What this PR does / why we need it? Fix function name typo, make `mask_fill_` to `masked_fill_` ### Does this PR introduce _any_ user-facing change? No ### How was this patch tested? CI passed Signed-off-by: ApsarasX --- vllm_ascend/attention/attention.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/vllm_ascend/attention/attention.py b/vllm_ascend/attention/attention.py index 38abe99..b8167fe 100644 --- a/vllm_ascend/attention/attention.py +++ b/vllm_ascend/attention/attention.py @@ -135,7 +135,7 @@ class AttentionMaskBuilder: context_len:] = self.splitfuse_mask_value right_tensor = attn_mask[current_row:current_row + q_len, context_len:seq_len] - right_tensor.mask_fill_( + right_tensor.masked_fill_( right_tensor.tril() == self.splitfuse_mask_value, 0) current_row += q_len