✨ ZZ-WO-20260906-002 · D248 Task 55 hc_head v2 修复 num_stages
之之 D248 00:13 反馈: v1 全 8 芯 Failed
参考 extend_attention.py v2 修复记录 (Liger-Kernel v2 经验), 头号嫌疑:
num_stages=1 显式 (v1 默认多级流水, 非 NVIDIA 后端支持不完整)
v1 → v2 改动 (从 extend_attention.py v2 学习):
1) num_stages=1 显式 (头号嫌疑)
2) num_warps 4 → 8 (大 BLOCK_D=1024 + HC_MULT=4 循环, 8 warps 更稳)
3) pid int64 → int32 (T < 2.1B, 长度算术用 int32 跟 extend_attention 一致)
4) 保留 v1 的指针 cast (D245 验证)
5) 保留 enable_fp_fusion=False (D245 验证)
v2 算法层 = v1 算法层 (参数不同, 逻辑一致)
v1 算法层 bit-exact 11/11 已 pass, v2 应该一致
打 zip: /Users/zhizhi/Desktop/hc_head.zip (内部 hc_head.py = v2 内容, 5215 bytes)
3 files, 1 commit
阿念 (Mavis, ICE-GL-AN-001) · Code · 之之的家 · 2026-09-06 D248 00:18 CST
da33d8536d