Closed casper2002casper closed 2 years ago
I cannot reproduce this error. Can you try to run the node and graph classification examples in the examples/
directory?
For me they work fine on gpu with the following package versions:
pkg> st
Status `~/.julia/dev/GraphNeuralNetworks/examples/Project.toml`
[052768ef] CUDA v3.8.1
[aae7a2af] DiffEqFlux v1.45.1
[0c46a032] DifferentialEquations v7.1.0
[587475ba] Flux v0.12.8
[7e08b658] GeometricFlux v0.8.0
[cffab07f] GraphNeuralNetworks v0.3.13
[3ebe565e] GraphSignals v0.3.9
[86223c79] Graphs v1.6.0
[eb30cadb] MLDatasets v0.5.15
[872c559c] NNlib v0.7.34
[a00861dc] NNlibCUDA v0.1.11
It's caused by using UInt16 node data. I might have used a different data type last time I tried using the GPU, so I guess that broke it for me. Could it be possible my GPU doesn't support 16-bit data types CUDA operations?
using GraphNeuralNetworks
using Flux
nn = GNNChain(GCNConv(2 => 2, relu))
x = rand_graph(5,10, ndata=rand(UInt16,2,5))
println("CPU")
println(nn(x))
println("GPU")
nn = Flux.gpu(nn)
x = Flux.gpu(x)
println(nn(x))
CPU
GNNGraph:
num_nodes = 5
num_edges = 10
ndata:
x => (2, 5)
GPU
ERROR: LoadError: Failed to compile PTX code (ptxas exited with code 255)
ptxas /tmp/jl_JXO7fH.ptx, line 314; error : Instruction 'atom.cas.b16.global' requires .target sm_70 or higher
ptxas fatal : Ptx assembly aborted due to errors
If you think this is a bug, please file an issue and attach /tmp/jl_JXO7fH.ptx
Seems like my GPU doesn't support 16 bit operations, based on https://docs.nvidia.com/cuda/cuda-c-programming-guide/index.html#maximize-instruction-throughput with my CC 5.0 GPU.
After updating to the latest version, GPU executing stopped working for me. I have not created a minimal reproducible example yet, but this is the error I'm facing.