What about providing C++ API in pyg, like that in `pytorch_cluster` ?

🚀 Feature

Providing C++ API, like that in pytorch_cluster, pytorch_scatter, pytorch_sparse, pytorch_spline_conv ?

Motivation

Graph convolution applications are gradually deployed in embedding devices. However, there might not be a proper . What’s more, a training procedure with pytorch+pytorch_geometric can be done easily, but the training procedure with libtorch+libtorchgeometric does not exist.

“From Research To Production”, the team provides PyTorch for research, and LibTorch for production. And since last year, GNN has been well developed in research area. Many new methods are proposed. And GNNs has been used in many applications. In my opinion, C++ API can make industrial deployment easier for some practical use.

Additional context

Perhaps this feature request is unrealistic, because as the project repositories shown, torch_geometric is much more like a front-end library which provides tools for research, and other 4 libraries (torch_cluster, torch_scatter, torch_sparse, torch_spline_conv) is the back-end libraries for the front-end. The researchers just know to write the proper Python code (pyg code), and do not need to concern C++/CUDA code. I know the reason why it only has Python code in pyg.

This is just my personal thought, and if other people think this is of no need, I will not mention that again.

Yours sincerely, @wmf1997

P.S. Another reason for this issue is that I want to write some examples for torch_cluster’s C++ API. See this issue

[rusty1s]

Awesome! Thanks for all your help. Let’s first make JIT work 😃

[liaopeiyuan]

Hi @rusty1s , I’ve been focusing on deploying graph neural networks in edge computing applications and production environment, so I maybe able to help with designing a C++ api for the package. I’m also familiar with PyTorch C++ API. I believe that maintaining a libtorchgeometric in parallel with the current PyTorch+Python api would be awesome and facilitate a lot in applying graph neural networks in industrial scenarios. You are always welcome to contact me about your ideas around July/August, and before then I will be mainly working on JIT support in Python.