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

从 CUDA 到 MLX:K-Search 如何把内核优化经验迁移到 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 到 MLX 翻译层,让已有 CUDA 内核作为知识库适配到 Apple Silicon。

推荐理由

原文给出了 CUDA 到 MLX 的翻译层设计与两组内核实测数据,可了解跨硬件迁移优化知识的具体做法。

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