search

JuliaGraphs / GraphNeuralNetworks.jl

Graph Neural Networks in Julia
https://juliagraphs.org/GraphNeuralNetworks.jl/graphneuralnetworks/
MIT License
227 stars 45 forks source link

graph NeuralODE example not working on gpu #56

Closed CarloLucibello closed 3 years ago

CarloLucibello commented 3 years ago

Set device=gpu in https://github.com/CarloLucibello/GraphNeuralNetworks.jl/blob/master/examples/neural_ode.jl to reproduce

CarloLucibello commented 3 years ago

This has been mostly fixed by #67. GPU performance seems good although there is still some scalar indexing done by one of the SciML packages.