Skip to content

Commit d6e9417

Browse files
authored
[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 <apsarax@outlook.com>
1 parent afe1767 commit d6e9417

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

vllm_ascend/attention/attention.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -135,7 +135,7 @@ def get_splitfuse_attn_mask(
135135
context_len:] = self.splitfuse_mask_value
136136
right_tensor = attn_mask[current_row:current_row + q_len,
137137
context_len:seq_len]
138-
right_tensor.mask_fill_(
138+
right_tensor.masked_fill_(
139139
right_tensor.tril() == self.splitfuse_mask_value, 0)
140140
current_row += q_len
141141

0 commit comments

Comments
 (0)