执行摘要
本次PR为jit_kernel模块的is_arch_support_pdl函数添加了@cache_once装饰器以实现结果缓存，并简化了cache_once装饰器的实现。这是一个小型性能优化和代码清理，风险较低，对用户透明，适合快速了解缓存装饰器的使用技巧。
功能与动机
为什么做：根据PR描述，主要目的是避免is_arch_support_pdl函数的重复计算，通过缓存其返回值来提升性能。同时，清理cache_once装饰器中冗余的key=lambda参数，使代码更简洁。PR body中明确写道：“Add @cache_once decorator to jit_kernel's is_arch_support_pdl so the result is cached - Remove redundant key=lambda from sorted(kwargs.items()) in cache_once”。
实现拆解
变更仅涉及一个文件python/sglang/jit_kernel/utils.py的两处修改：
- 添加缓存装饰器：在
is_arch_support_pdl函数定义前添加@cache_once，使该函数的结果在首次调用后被缓存。
python
@cache_once
def is_arch_support_pdl() -> bool:
if is_hip_runtime():
return False
- 简化装饰器实现：修改
cache_once内部的key生成逻辑，从tuple(sorted(kwargs.items(), key=lambda x: x[0]))简化为tuple(sorted(kwargs.items()))，因为sorted默认按字典键排序，key=lambda参数是冗余的。
评论区精华
本次PR没有review讨论记录（review_comments_count为0），作者merrymercy自行合并，表明变更简单直接，未引发技术争议。
风险与影响
风险：
- 缓存一致性：
is_arch_support_pdl被缓存后，如果运行时硬件架构动态变化，缓存结果可能不会更新，但该函数检测的是静态架构特性，实际风险较小。
- 代码简化副作用：移除
key=lambda理论上不影响功能，但如果未来kwargs包含非标准键，排序行为可能有细微差异。
- 测试覆盖不足：仅依赖现有CI测试，未新增针对缓存行为的单元测试。
影响：
- 性能：减少重复硬件检测开销，对启动或运行时性能有轻微提升。
- 代码质量：简化装饰器实现，提高可读性。
- 用户：透明无感，不改变API或功能。
关联脉络
从近期历史PR看，本PR与以下JIT内核相关PR存在关联：
- PR #20673：为Minimax模型引入融合TP QK norm JIT内核，同属
jit-kernel模块的性能优化。
- PR #22155和#22187：涉及hisparse JIT内核的测试和基准测试，共享相同模块上下文。
这些PR共同反映了jit-kernel模块在持续进行性能优化和测试完善，本PR的小型缓存优化是该趋势的一部分。
参与讨论