跳到正文
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 内核经验如何被结构化迁移到 Apple Silicon,以及翻译层在性能提升中的具体作用。

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