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

UC Berkeley 与 IBM 扩展 K-Search:将 CUDA 算子优化经验自动迁移至 Apple Silicon MLX

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

AI 导读

UC Berkeley Sky Lab 与 IBM Research 研究人员扩展了进化式算子搜索框架 K-Search,通过构建结构化 CUDA-to-MLX 翻译层,将成熟的 CUDA 优化经验自动迁移至 Apple Silicon 的 MLX 框架。

推荐理由

该工作展示了如何通过结构化映射层将成熟算子优化规则迁移到新芯片,为非英伟达硬件生态自动生成高性能算子提供了可复用路径。

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