变更摘要

本 PR 主要针对稀疏/量化 MLA 测试链路做 bugfix，围绕三方面：修复 golden 参考实现中 ori_mask_mode == 3 等“行无效”场景的判断逻辑、修复测试用例“精度失败却被判为 Pass”的问题，以及为 return_softmax_lse 场景补齐 LSE 输出。核心改动集中在 *_golden.py 参考实现与各测试入口文件：移除了提前置零并 continue 的旧逻辑，改为在窗口为空时构造空张量并仅在 cur_ori_k_bnsd 与 cur_cmp_k 均为空时才将输出行置零；同时调整了 sinks_softmax 的返回值以支持 LSE 计算，并将多个测试文件的失败判定从 == "Failed" 等改为 != "Pass"。

主要改动

• Golden 行无效逻辑修复：mixed_quant_sparse_flash_mla_golden.py、quant_sparse_flash_mla_golden.py 与 sparse_flash_mla_golden.py 均删除了按 ori_mask_mode == 3 提前置零并 continue 的旧逻辑，改为在 ori_win_start >= ori_win_end 时构造空的 cur_ori_k_bnsd，仅在 cur_ori_k_bnsd 与 cur_cmp_k 都为空时才将 attn_out 对应行置零并 continue。
• LSE 计算适配：mixed_quant_sparse_flash_mla_golden.py 中 sinks_softmax 的返回值由 (y, x_sum) 改为 (y, x_max, x_sum)，并在 return_softmax_lse 为真时向 softmax_lse 写入 x_max[:, 0] + torch.log(softmax_sum[:, 0] + 1e-10)。
• 空窗口拼接与循环次数修复：sparse_flash_mla_golden.py 中 ori_win_end 改为 max(ori_threshold, 0)，并针对 cur_ori_k_bnsd 或 cur_cmp_k 单侧为空的情况调整了 k_concat 的拼接方式；当 empty_flag_ori 时 total_s2_loop_time 由 0 改为 cmp_s2_loop_time。
• 测试失败判定修复：test_mixed_quant_sparse_flash_mla_batch.py、test_mixed_quant_sparse_flash_mla_batch_graph.py、test_mixed_quant_sparse_flash_mla_single.py 及 test_sparse_flash_mla_batch_graph.py 将 result == "Failed" 的判断改为 result != "Pass"，test_quant_sparse_flash_mla_batch.py 与 test_quant_sparse_flash_mla_single.py 将 result in ("NPU ERROR", "Failed") 改为 result != "PASS"，避免非通过结果被漏判。
• 图模式测试参数化补全：test_mixed_quant_sparse_flash_mla_batch_graph.py 与 test_sparse_flash_mla_batch_graph.py 补充了 testcase_ids 列表，并对环境变量读取、引号风格等进行了统一整理。

compile

/approve