Examples on GPU

[navidcy]

Some of the examples need slight modifications/generalizations to be able to run on GPU by simply setting dev=GPU().

[navidcy]

This issue was brought up by @ranocha in his review.

The main reason for examples failing is plotting: when plotting one needs to bring the arrays on CPU. E.g.,

heatmap(x, y, ζ')

won't work if ζ is a CuArray and instead we need to call

heatmap(x, y, collect(ζ)')

@glwagner any ideas what we can do here? Sprinkle collect() in all plotting commands in examples?

[navidcy]

Or we can overload some plotting functions when we load GeophysicalFlows.jl, e.g.

import Plots.plot, Plots.heatmap, Plots.contourf

Plots.plot(x, y::CuArray, args...; kwargs...) = Plots.plot(x, collect(y), args...; kwargs...)
Plots.heatmap(x, y, z::CuArray, args...; kwargs...) = Plots.heatmap(x, y, collect(z), args...; kwargs...)
Plots.contourf(x, y, z::CuArray, args...; kwargs...) = Plots.contourf(x, y, collect(z), args...; kwargs...)

[navidcy]

hm... no, we don't want to do that in GeophysicalFlows.jl
perhaps within the examples?