cuda_kernels help

[tchaton]

Hey,

I am working deeply on your code.
I would like to ask you a favor, and if you could please help me to understand the cuda kernels.
My email adress is thomasc@helix.re

I have benchmark your code vs pytorch geometric own implementation and I would say yours is approximately 3 times faster to train.

Best,

[mys007]

That's funny to hear. Can you perhaps share your benchmark code?

[tchaton]

The conversation is there: pyg-team/pytorch_geometric#602

[tchaton]

I am also part of Helix and work with Loic on spg3.0 :)

[tchaton]

pyg-team/pytorch_geometric#602
Based on this one, what s missing is the modification made in learning/main.py.
If needed, I would provide you with them :)

[mys007]

Thanks and sorry for the delay! I've just replied to the pytorch_geometric thread, let's continue there.