dmlc/dgl

Support TorchScript for DGL Sparse

Open

#5 275 ouverte le 9 févr. 2023

Voir sur GitHub
 (3 commentaires) (0 réactions) (1 assigné)Python (2 928 forks)batch import
feature requesthelp wanted

Métriques du dépôt

Stars
 (12 665 stars)
Métriques de merge PR
 (Aucune PR mergée en 30 j)

Description

Description

There have been many requests to have DGL support JIT compilation. Unfortunately, it is difficult to JIT DGL's message passing API because it supports very flexible user-defined message and reduce functions. On the other hand, adding such support to DGL Sparse is relatively easier because DGL Sparse API is closer to PyTorch's Tensor API.

This issue is to track the progress of such support and list the encountered problems for open discussions.

Findings

Work items

  • #5412
  • #5413
  • #5414
  • #5415
  • #5416

Guide contributeur

Direction de recherche
Examinez l'API DGL Sparse et comprenez les contraintes de TorchScript pour identifier les modifications nécessaires à la compilation JIT.
Stack technique
pythonpytorch
Domaine
machine learningai
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.
2
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
PythonPyTorch
Accessibilité débutantScore de 1 à 100 estimant l'accessibilité de cette issue pour un premier contributeur.
50