ParallelKernelBench:前沿 LLM 尚无法编写快速多 GPU 内核
Together AI 推出含 87 道真实代码问题的 ParallelKernelBench,测试前沿 LLM 能否用经 NVLink 直接传输数据的 CUDA 内核替代 PyTorch + NCCL。最佳模型零样本仅正确解决 28 题,其中 22 个方案更快;采样三次时 fast 1@3 最高仍为 31%。智能体反馈将 Gemini 3 Pro 的正确数从 24 提升至 35,但约 20 轮后停滞,主要难点仍是 rank 协调、通信顺序和传输机制选择。
ready accepts Triton and ParallelKittens solutions, and emerging interfaces like NCCL GIN and NVSHMEM are worth studying as targets. Expanding support to these paradigms will encourage further research into how AI agents navigate diverse programming models and hardware abstractions. The broader goal is a concrete target for the harder problem behind all of this: LLM systems that can autonomously optimize and manage large-scale distributed infrastructure. For infrastructure custom-built for langu
匹配字段正文









