dmlc/dgl

Support TorchScript for DGL Sparse

オープン

#5,275 opened on 2023/02/09

 (3 件のコメント) (0 件のリアクション) (1 人の担当者)Python (2,928 件のフォーク)batch import
feature requesthelp wanted

Repository metrics

Stars
 (12,665 個のスター)
PR merge metrics
 (PR metrics pending)

説明

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

コントリビューターガイド