Kimi-K3 GPU JPEG 解码改走 nvImageCodec fancy 上采样,精度与 PIL 对齐
值得精读。核心看点有三个:一是用第三方 nvImageCodec 替换 torchvision 默认 nvJPEG 行为的思路——通过 `fancy_upsampling` 配置解决 4:2:0 色度重建精度问题的根因定位非常清晰;二是 decoder 池的容量上限设计(2 实例 + LIFO + 创建锁)在并发与 HBM 之间取平衡;三是测试用 fake `nvidia.nvimgcodec` 模块在纯 CPU 环境验证 GPU 解码逻辑,避免 CI 依赖 GPU,是很好的 mock 模式。若你的团队也在做视觉预处理精度对齐,这份 PR 的验证方法论(paired per-sample score matrix + raw-pixel MAE 基准)可以直接复用。
参与讨论