NVIDIA / NVIDIA/nvmath-python

Triton JIT support for device APIs

Aperta
#16 3 commenti 0 reazioni 0 assegnatari Vedi su GitHub

Nessuno ha ancora preso questa issue.

Lingua principale
Cython
Stelle
601
Fork
46
Metriche di merge delle PR
Nessuna PR unita negli ultimi 30g

Descrizione

There is already support for Numba CUDA JIT but Triton JIT interacts much better with frameworks like PyTorch.

Guida per i contributori

Apri la guida per i contributori

Come iniziare

  1. Leggi tutta la issue e poi la guida ai contributi del progetto.
  2. Commenta sulla issue per dire che te ne occupi tu — evita che due persone facciano lo stesso lavoro.
  3. Fai un fork del repository e lavora su un branch.
  4. Apri una pull request che faccia riferimento al numero della issue.

Direzione di ricerca

L’issue non indica file, test o punti di ingresso. Inizia individuando il supporto esistente per Numba CUDA JIT e verificando come vengono esposte le API del dispositivo, quindi confronta i requisiti di Triton JIT con PyTorch. Il lavoro sarà completato quando il supporto per Triton JIT sarà disponibile per le API del dispositivo pertinenti, con una copertura dei casi d’uso supportati.

Scritto dal modello di indicizzazione a partire dal testo della issue.

Valutazione

Stack tecnologico
python, pytorch
Ambito
backend, machine-learning
Tipo di issue
Funzionalità
Difficoltà
5/5
Tempo stimato
Più di una settimana
Stato di attività
Ferma
Chiarezza
Da chiarire
Idoneità per principianti
25/100

Ricevi le nuove issue nella tua casella

Un breve riepilogo di issue GitHub adatte ai principianti.