Support TorchScript for DGL Sparse
Open
feature request
help wanted
- Dominant language
- Python
- Stars
- 14.3k
- Forks
- 3.1k
- PR merge metrics
- No merged PRs in 30d
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**
* Some APIs need to be changed (like https://github.com/dmlc/dgl/commit/be3b0e6127446c26d4d157d5746f9e82e3a7cc02).
## Work items
* #5412
* #5413
* #5414
* #5415
* #5416
Contributor guide
No contributing guide indexed for this repository
Assessment
This issue has not been assessed yet.