Skip to content

Commit 6ee9dc8

Browse files
committed
BSND to TND and FA_UPDATE replacement
Signed-off-by: pichangping <[email protected]>
1 parent 0d67a17 commit 6ee9dc8

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

vllm_ascend/attention/attention_v1.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -973,6 +973,7 @@ def _forward_decode_pcp_dcp(self, query: torch.Tensor,
973973
# Update out&lse
974974
attn_out = self._npu_attention_update(attn_out_lse_list)
975975
return attn_out
976+
976977
def _forward_pcp_dcp(self, query: torch.Tensor, key: torch.Tensor,
977978
value: torch.Tensor, attn_metadata: AscendMetadata,
978979
output: torch.Tensor) -> torch.Tensor:

0 commit comments

Comments
 (0)