执行摘要

• 一句话：修复CPU测试因导入路径变更导致的NotImplementedError，恢复使用原生CPU实现。
• 推荐动作：这是一个简单的修复PR，无需深入精读。值得关注的是它揭示了测试环境对硬件依赖的敏感性，以及导入路径选择对跨平台兼容性的影响。

功能与动机

修复由PR #20796引入的CPU CI失败问题。PR #20796将测试导入路径改为使用社区版fa3内核，但该路径在CPU环境下会检查CUDA sm80+支持，导致抛出NotImplementedError。需要恢复使用CPU原生实现以确保测试通过。

实现拆解

仅修改一个测试文件test/srt/cpu/test_flash_attn.py：

1. 移除对sglang.jit_kernel.flash_attention的导入；
2. 添加sgl_kernel导入（仅用于类型提示）；
3. 将flash_attn_varlen_func直接赋值为torch.ops.sgl_kernel.flash_attn_varlen_func，这是CPU原生实现。

关键文件：

• test/srt/cpu/test_flash_attn.py（模块 test）: 唯一被修改的文件，修复了导入路径导致的CPU测试失败问题。

关键符号：flash_attn_varlen_func

评论区精华

本PR没有review评论，但从PR body和关联Issue可以看出，这是一个直接的修复，旨在解决由PR #20796引入的回归问题。

• 导入路径导致的CPU测试失败 (correctness): 恢复使用torch.ops.sgl_kernel.flash_attn_varlen_func作为CPU原生实现。

风险与影响

• 风险：风险极低：
  1. 仅修改测试文件，不影响生产代码；
  2. 恢复为原有工作路径，已被历史验证；
  3. 变更范围极小（4行改动），回归风险可控。
• 影响：影响范围有限：
  1. 确保CPU CI测试套件（per-commit-cpu）能够正常执行test_flash_attn.py；
  2. 对用户和系统无直接影响；
  3. 对团队而言，修复了CI稳定性问题，避免因测试失败干扰开发流程。
     - 风险标记：测试环境依赖

关联脉络

• PR #20796 Kernels community fa3: 本PR修复了由#20796引入的回归问题，#20796将导入路径改为使用社区版fa3内核，导致CPU测试失败。