search

CarloLucibello / GraphNeuralNetworks.jl

Graph Neural Networks in Julia
https://carlolucibello.github.io/GraphNeuralNetworks.jl/dev/
MIT License
209 stars 47 forks source link

Support AMDGPU #343

Open CarloLucibello opened 9 months ago

CarloLucibello commented 9 months ago

Needs to implement gather/scatter kernels in NNlib

CarloLucibello commented 9 months ago

Gather/scatter functionality has been already implemented in https://github.com/FluxML/NNlib.jl/pull/487. Someone with an amd gpu should test one of the layers here and see if everything works fine.