变更摘要

此 PR 修复了 SK（StreamK）算法中单核切 K 尾块的计算问题。主要将 FP32 场景下 split K 的触发判断从 BlockScheduler 构造函数移至 UpdateBlockK（原 UpdateSplitKIdx）方法中，并将判断条件从原有的 skKSingleCore_ > splitSingleK_ 改为基于常量阈值 FP32_SPLIT_K_THRESHOLD（2048） 的 realSingleCoreK_ >= FP32_SPLIT_K_THRESHOLD，同时为 splitSingleKRound_ 增加了 realSingleCoreK_ < splitSingleK_ 时的兜底保护，避免异常计算。调用方 kernel_matmul_streamk.h 同步更新了方法名。

主要改动

• Split K 触发逻辑延迟到运行时判断：将 isSplitSingleK_ 的设置从 BlockScheduler 构造函数移至 UpdateBlockK（原 UpdateSplitKIdx）方法中，确保判断时 realSingleCoreK_ 等运行时参数已就绪，避免构造阶段使用未正确初始化的值导致尾块计算错误。
• Split K 判断条件改为固定阈值：引入常量 FP32_SPLIT_K_THRESHOLD = 2048，将原判断条件 skKSingleCore_ > splitSingleK_ 替换为 realSingleCoreK_ >= FP32_SPLIT_K_THRESHOLD，使 split K 的触发不依赖于动态的 splitSingleK_，消除因参数变化导致的尾块划分异常。
• splitSingleKRound_ 增加兜底保护：将 splitSingleKRound_ 的计算从 realSingleCoreK_ / splitSingleK_ 改为 realSingleCoreK_ < splitSingleK_ ? 1 : (realSingleCoreK_ / splitSingleK_)，防止 realSingleCoreK_ 小于 splitSingleK_ 时出现除零或结果为 0 的异常情况。
• 方法重命名：UpdateSplitKIdx 重命名为 UpdateBlockK，kernel_matmul_streamk.h 中的调用方同步更新，使方法名更准确地反映其职责（更新 block K 相关状态而非仅更新 split K 索引）。

代码审查

审查总结

审查结果

问题统计

总体风险评估

高风险。P0 问题会导致 FP32 非 HF32 连续格式且 K ≥ 2048 场景下，split-K 的 C 矩阵累加失去原子保护，多 K 分片并发写入同一位置时产生数据竞争，计算结果不可信。P1 问题在 DP tile 与 SK tile 的 realSingleCoreK_ 跨越阈值时（如 k_ >= 2048 但 skKSingleCore_ < 2048）会触发错误的 K 分片参数计算。两个问题均需在上线前修复。

⛔ 需要修改

CLA Signature Pass

pzyy00, thanks for your pull request. All authors of the commits have signed the CLA. 👍

The following users do not have permission to comment /lgtm or /approve on any module in this PR:
商晓波