FluxML / GeometricFlux.jl

Geometric Deep Learning for Flux
https://fluxml.ai/GeometricFlux.jl/stable/
MIT License
348 stars 30 forks source link

Message-passing for CUDA is broken by not allowing scalar indexing #195

Closed yuehhua closed 2 years ago

yuehhua commented 3 years ago

Mentioned in #187.

Message-passing network has an operation of scalar indexing over a cuarray. Currently, new version of CUDA forbidden the use of scalar indexing, which breaks the tests. Message-passing need a new implementation for indexing vertices or edges, maybe a cuda kernel.

yuehhua commented 2 years ago

Now it is fixed by #253