Skip to content

Commit 222ea13

Browse files
author
zhangyue
committed
fix(ascend): remove stale 910B skip guard from PagedAttention test
The skip was based on an outdated diagnosis that ATB PagedAttention crashes during Setup on 910B + CANN 8.5.x. After the framework rebase onto master (which includes the pybind11 kw arg order fix), all 10 parametrizations pass on 910B4 with CANN 8.5.1. Keep the NPU-available and implementation-registered checks since they are cheap, structural prerequisites.
1 parent 146dc8d commit 222ea13

1 file changed

Lines changed: 0 additions & 8 deletions

File tree

tests/test_paged_attention.py

Lines changed: 0 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -18,14 +18,6 @@ def _atb_pa_unsupported_reason():
1818
if not infini.ops.PagedAttention.active_implementation_indices("ascend"):
1919
return "ATB PagedAttention implementation not registered for Ascend"
2020

21-
# ATB PA crashes during `Setup` on Ascend 910B (CANN 8.5.x). Other
22-
# SoCs (Atlas A5 SoC 260) are known to work. Extend the blacklist as
23-
# more bad SoCs are identified.
24-
name = torch.npu.get_device_name(0)
25-
26-
if "910B" in name:
27-
return f"ATB PagedAttention crashes on {name} with CANN 8.5.x"
28-
2921
return ""
3022

3123

0 commit comments

Comments
 (0)