#22725 [Misc] Use cache_once for is_arch_support_pdl in sgl-kernel
作者 merrymercy · 合并时间 2026-04-15 06:22
为 sgl-kernel 的架构检测函数添加缓存装饰器并增强健壮性。
该 PR 值得精读,尤其是 `cache_once` 装饰器的实现,它展示了如何为 `torch.compile` 兼容性设计轻量级缓存。关注点包括:装饰器的键设计、异常处理策略、以及如何统一跨模块的架构检测逻辑。对于涉及 PyTorch 编译或跨平台 GPU 代码的开发者,这是一个实用的参考案例。
参与讨论