ml-explore/mlx

[Feature Request] Fallback for custom kernels

已关闭

#3,240 创建于 2026年3月11日

在 GitHub 查看贡献者指南
 (4 条评论) (0 个反应) (0 位负责人)C++ (1,794 个派生)batch import
enhancementgood first issue

仓库指标

星标
 (26,216 个星标)
PR 合并指标
 (平均合并 6天 17小时) (30 天内合并 75 个 PR)

描述

We have a specific use-case whereby we need to export MLX callables from Linux. Currently, this prevents us from using any operations that require custom kernels (e.g. grid sample) because otherwise, MLX complains that no Metal GPU was found.

Ideally, we would want to keep the fast kernel so that we can export without a Metal GPU, but execute on a Metal GPU.

贡献者指南

研究方向
研究MLX中自定义内核的编译和执行方式，然后设计一个回退机制，当Metal GPU不可用时，使用软件实现或基于CPU的内核。重点关注网格采样操作及类似的自定义内核。
技术栈
无
领域
machine learningbackend
议题类型
功能
难度面向新贡献者的预计实现难度，1 表示很小改动，5 表示专家级工作。
3
预计时间有经验贡献者完成调查、实现、测试并准备 pull request 的粗略时间范围。
1-2 天
活动状态议题当前的可参与程度：新鲜、活跃、陈旧、阻塞或等待维护者输入。
活跃
清晰度议题是否清楚说明期望改动、验收标准和下一步。
清晰
前置要求
C++
新手友好度1-100 的估计分数，表示该议题对首次贡献者的友好程度。
30