dmlc--dgl
4208ce2b9e
Co-authored-by: Zihao Ye <expye@outlook.com>
553 B
553 B
FeatGraph-DGL
FeatGraph is an efficient backend for Graph Neural Networks based on TVM.
- Original repo: https://github.com/amazon-research/FeatGraph
- SC2020 Paper: https://www.csl.cornell.edu/~zhiruz/pdfs/featgraph-sc2020.pdf
This folder contains the code for integrating featgraph kernels to DGL.
Usage
After building DGL with USE_TVM=ON, you should be able to run:
python test.py
to verify correctness.