跳到正文
热点事件历史事件

K-Search实现CUDA到MLX内核自动迁移

1 篇报道1 个报道来源2026-07-29更新

先了解这件事

报道摘要

Berkeley AI Research发布博客介绍K-Search框架的新进展。该框架基于进化搜索,利用AI优化GPU内核。团队扩展了MLX后端,并开发了一种结构化的CUDA到MLX翻译层,使系统能自动将现有的CUDA内核知识迁移至Apple Silicon芯片,而非从零重建。实验显示,该方法在Apple Silicon上达到接近专家水平的性能,相比原生MLX Attention内核加速比为0.97x,预填充速度最高提升20倍。

摘自 BAIR:Berkeley AI Research Blog

报道时间线

沿着报道,了解事件的不同侧面。

7月29日
  1. BAIR:Berkeley AI Research Blog
    K-Search框架实现CUDA内核向Apple Silicon MLX的自动迁移与优化

    Berkeley AI Research发布博客介绍K-Search框架的新进展。该框架基于进化搜索,利用AI优化GPU内核。团队扩展了MLX后端,并开发了一种结构化的CUDA到MLX翻译层,使系统能自动将现有的CUDA内核知识迁移至Apple Silicon芯片,而非从零重建。实验显示,该方法在Apple Silicon上达到接近专家水平的性能,相比原生MLX Attention内核加速比为0.97x,预填充速度最高提升20倍。