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