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

从 CUDA 到 MLX:K-Search 如何把内核优化经验迁移到 Apple Silicon

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

AI 导读

IBM Research 基于 UC Berkeley Sky Lab 的 K-Search 框架新增 MLX 后端,并设计结构化 CUDA-to-MLX 翻译层,让已有 CUDA 内核作为知识库被改写为 Apple Silicon 内核。

推荐理由

原文给出了从 CUDA 到 MLX 的翻译层设计与两组内核实测数据,可看跨平台内核经验迁移的具体做法。

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