Closed liasiegelman closed 2 years ago
Can you post your whole script?
The params seem to include a matrix. Probably that’s the issue.
Can you print typeof(prob.params.eta)
and typeof(prib.params.etah)
?
You are correct Navid, thanks ! I failed to create a CUDA array somehow. Will fix this after a surf ! Thanks :)
I’m opening it to attempt to make this user friendlier.
Sure maybe there's a way to implement an error message such that when dev=GPU(), and eta is not a cuda array it returns an error message when constructing the problem (prob = SingleLayerQG.Problem(dev; nx=n, Lx=L, eta=η, deformation_radius=deformation_radius, dt=dt, stepper=stepper))
Was thinking the constructor would convert it to a CuArray if dev is GPU.
When creating a diagnostic for enstrophy I get the following error (no problem for KE and PE)