ml-explore/mlx

[Feature Request] Fallback for custom kernels

Fermée

#3 240 ouverte le 11 mars 2026

Voir sur GitHubGuide contributeur
 (4 commentaires) (0 réaction) (0 personne assignée)C++ (1 794 forks)batch import
enhancementgood first issue

Métriques du dépôt

Stars
 (26 216 étoiles)
Métriques de merge PR
 (Merge moyen 6j 17h) (75 PRs mergées en 30 j)

Description

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.

Guide contributeur

Direction de recherche
Analysez comment les noyaux personnalisés sont compilés et exécutés dans MLX, puis concevez un mécanisme de repli qui utilise une implémentation logicielle ou des noyaux basés sur CPU lorsque le GPU Metal n'est pas disponible. Concentrez vous sur l'opération d'échantillonnage de grille et les noyaux personnalisés similaires.
Stack technique
Aucun
Domaine
machine learningbackend
Type d'issue
Fonctionnalité
DifficultéDifficulté estimée pour un nouveau contributeur, de 1 pour un très petit changement à 5 pour un travail expert.
3
Temps estiméFourchette de temps approximative pour investiguer, implémenter, tester et préparer une pull request.
1-2 jours
Statut d'activitéDisponibilité apparente de l'issue : fraîche, active, ancienne, bloquée ou en attente d'un mainteneur.
Active
ClartéClarté avec laquelle l'issue explique le changement attendu, les critères d'acceptation et la prochaine étape.
Claire
Prérequis
C++
Accessibilité débutantScore de 1 à 100 estimant l'accessibilité de cette issue pour un premier contributeur.
30