dmlc/dgl

Support TorchScript for DGL Sparse

Aberta

#5.275 aberto em 9 de fev. de 2023

Ver no GitHub
 (3 comentários) (0 reação) (1 responsável)Python (2.928 forks)batch import
feature requesthelp wanted

Métricas do repositório

Stars
 (12.665 estrelas)
Métricas de merge de PR
 (Métricas PR pendentes)

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

Guia do colaborador

Direção de pesquisa
Examine a API DGL Sparse e entenda as restrições do TorchScript para identificar as alterações necessárias para a compilação JIT.
Pilha de tecnologia
pythonpytorch
Domain
machine learningai
Tipo Issue
Funcionalidade
DifficultyDificuldade de implementação estimada para um novo contribuidor, de 1 para alterações muito pequenas a 5 para trabalho de nível especializado.
2
Tempo estimadoUm intervalo de tempo aproximado para um colaborador experiente investigar, implementar, testar e preparar um pull request.
1-2 dias
Status da atividadeQuão disponível o issue aparece agora: novo, ativo, obsoleto, bloqueado ou aguardando entrada do mantenedor.
Ativo
ClarityCom que clareza o issue explica a mudança esperada, os critérios de aceitação e a próxima etapa.
Claro
Prerequisites
PythonPyTorch
Simpatia para novatosUma pontuação de 1 a 100 que estima o quão acessível este issue é para colaboradores iniciantes.
50