GPU usage of drop edge

[songyesog2000]

Sorry for asking the dumb question here since i am a newbee for programming. I met a slow running on GPU using drop edge in each layer of my nn model. Drop edge documentation here. I wonder is this problem caused by the cuda compatibility of the function or the generation of the random bool value.

Thank you in advance for checking this for me and maintaining such an amazing package!
Sam

[rusty1s]

Thanks for reporting. There were indeed some issues that slowed down GNN execution (in particular when force_undirected=True set). I fixed this in #4059.

[songyesog2000]

Thank you, really helpful!