跳到正文
Berkeley AI Research·精选AI 评分63

研究者用 K-Search 将 CUDA kernel 优化经验迁移到 Apple Silicon 的 MLX

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

AI 导读

IBM Research 基于加州大学伯克利 Sky Lab 的 K-Search 演进式 kernel 搜索框架,为其新增 MLX 后端和结构化 CUDA 到 MLX 翻译层,把概念映射表、MLX 专有模式和可复用断言作为上下文供进化搜索使用。

推荐理由

原文给出可复现的 CUDA 到 MLX 翻译层做法与分项数据,读者可据此评估跨平台 kernel 迁移的方法是否适用。

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