跳到正文
Berkeley AI Research·· 2026-07-29精选AI 评分60

K-Search 从 CUDA 迁移到 MLX,为 Apple Silicon 生成高性能内核

From CUDA to MLX: How K-Search Brings Decades of Kernel Expertise to Apple Silicon

AI 导读

Berkeley Sky Lab 与 IBM Research 将 K-Search 内核搜索框架扩展到 MLX,通过结构化的 CUDA-to-MLX 翻译层让已有 CUDA 内核经验适配 Apple Silicon。

推荐理由

读者可以看到 CUDA 内核经验如何被结构化翻译成 MLX 优化策略,以及翻译层对最终性能的具体贡献。

来源:Berkeley AI Research · bair.berkeley.edu