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