There’s no reason to allocate imoutput for every iteration. It’s also performing a copy, where you should just construct a empty CuArray. But ultimately there should be some integration between the plotting library and CUDA.jl so that the array can be used directly, and that is currently not implemented.
2 Likes