[CPU] Support head_size 512 in cpu_attn (#38676)

Signed-off-by: jiang1.li <jiang1.li@intel.com>
2026-04-01 13:42:27 +08:00
parent 2d725b89c5
commit 36d7f19897
4 changed files with 4 additions and 4 deletions
--- a/tests/kernels/attention/test_cpu_attn.py
+++ b/tests/kernels/attention/test_cpu_attn.py
@@ -25,7 +25,7 @@ NUM_HEADS = [
    (8, 2),
    (9, 3),
 ]
-HEAD_SIZES = [96, 128]
+HEAD_SIZES = [96, 128, 512]
 HEAD_SIZES_VEC16 = [96, 80, 112, 128]
 QTYPES = [torch.bfloat16, torch.half, torch.float32]
 SLIDING_WINDOWS = [None, 256]