dmlc/dgl

Support TorchScript for DGL Sparse

Open

#5.275 geöffnet am 9. Feb. 2023

Auf GitHub ansehen
 (3 Kommentare) (0 Reaktionen) (1 zugewiesene Person)Python (2.928 Forks)batch import
feature requesthelp wanted

Repository-Metriken

Stars
 (12.665 Stars)
PR-Merge-Metriken
 (Keine gemergten PRs in 30 T)

Beschreibung

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

Contributor Guide

Research-Richtung
Untersuchen Sie die DGL Sparse API und verstehen Sie die Einschränkungen von TorchScript, um die erforderlichen Änderungen für die JIT Kompilierung zu identifizieren.
Tech Stack
pythonpytorch
Domain
machine learningai
Issue Type
Funktion
SchwierigkeitGeschätzte Implementierungsschwierigkeit für neue Contributors, von 1 für sehr kleine Änderungen bis 5 für Expertenarbeit.
2
Geschätzte ZeitEin grober Zeitrahmen, um zu recherchieren, zu implementieren, zu testen und einen Pull Request vorzubereiten.
1-2 Tage
AktivitätsstatusWie verfügbar das Issue gerade wirkt: frisch, aktiv, stale, blockiert oder wartet auf Maintainer-Input.
Aktiv
KlarheitWie klar das Issue die erwartete Änderung, Akzeptanzkriterien und den nächsten Schritt erklärt.
Klar
Voraussetzungen
PythonPyTorch
EinsteigerfreundlichkeitEin Score von 1 bis 100, der schätzt, wie zugänglich dieses Issue für First-Time Contributors ist.
50