Flux model with `SubArray` parameters don't work on GPU

It works now. I just tried it.

Many of the packages involved have been updated since I created the topic (4 months ago). Most notably, CUDA.jl went from version 3.1.0 to 3.3.4.
I haven’t touched the project in the meanwhile, so I don’t know which update solved the issue.

Thank you all for the help.

1 Like