ecc icon indicating copy to clipboard operation
ecc copied to clipboard

Published 20 hours ago • verified publisher icon mys007

cuda_kernels help

Open tchaton opened this issue 5 years ago • 5 comments

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 [email protected]

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

Best,

tchaton avatar Aug 06 '19 19:08 tchaton

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

mys007 avatar Aug 06 '19 21:08 mys007

The conversation is there: https://github.com/rusty1s/pytorch_geometric/issues/602

tchaton avatar Aug 07 '19 14:08 tchaton

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

tchaton avatar Aug 07 '19 14:08 tchaton

https://github.com/rusty1s/pytorch_geometric/issues/602 Based on this one, what s missing is the modification made in learning/main.py. If needed, I would provide you with them :)

tchaton avatar Aug 07 '19 14:08 tchaton

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

mys007 avatar Aug 21 '19 22:08 mys007