UC Berkeley 提出 K-search:将 CUDA 内核优化自动迁移至苹果 MLX

berkeley_ai · x · 2026-07-30

随着 AI 硬件生态日益碎片化,将成熟平台(如 NVIDIA CUDA)的内核优化经验迁移到新硬件(如 Apple Silicon)往往需要耗费大量重写精力。

UC Berkeley 与 IBM Research 的团队联合提出了 K-search 框架(演化式内核搜索)。该工具能将 CUDA 中积累的数十载优化知识自动转化为适配苹果 MLX 框架的原生策略。实测显示,其翻译出的注意力机制代码速度达到了苹果原生实现的 0.97 倍。项目已开源。

原文链接 →

「Infra」频道最新

更多「Infra」频道 AI 资讯 →