NetCDFOutputWriter not working in previous code after I update oceananigans

[rafacmartins]

Hi all,
So, last week I managed, somehow, to break the environment that I was working on, and then I had to reinstall Julia and Oceananigans. After reinstalling Oceananigans and Julia, I tried to run my previous code and found a problem in creating the netCDF output. Looking at the documentation, I found that the previous function NetCDFOutputWriter is now NetCDFWriter, then I changed it and tried to run the cod,e but now I'm receiving the following error message:

Scalar indexing is disallowed.
Invocation of getindex resulted in scalar indexing of a GPU array.
This is typically caused by calling an iterating implementation of a method.
Such implementations *do not* execute on the GPU, but very slowly on the CPU,
and therefore should be avoided.

If you want to allow scalar iteration, use `allowscalar` or `@allowscalar`
to enable scalar iteration globally or for the operations in question.

Stacktrace:
  [1] error(s::String)
    @ Base ./error.jl:35

I know this is a pretty common error in Oceananigans when using the GPU() architecture, but I hadn't encountered this issue before updating Oceananigans and Julia. The code I'm running for the output section is:

#Preparing the output
b, = simulation.model.tracers
u, v, w = simulation.model.velocities
ζ = ∂x(v) - ∂y(u)
div = ∂x(u) + ∂y(v)

extra_outputs = (;
    u=@at((Center, Center, Center), u),
    v=@at((Center, Center, Center), v),
    w=@at((Center, Center, Center), w),
    ζ=@at((Center, Center, Center),ζ),
    div = @at((Center, Center, Center),div),
)
outputs = merge(model.tracers,extra_outputs);

# setting folder name
name = "Simulation_Output"
output_folder = name

# Creates the folder if it doesn't exist
mkpath(output_folder)

println("Folder created: $output_folder")

# Setting the NetCDFWriter
simulation.output_writers[:fields] = NetCDFWriter( #here used to be NetCDFOutputWriter
    model,
    outputs,
    overwrite_existing=true,
    filename=name*".nc",
    dir = output_folder,
    schedule=TimeInterval(1hour)
    #global_attributes = Dict("U0" => U0 , "delta0" => delta0,"w_mask" => w_mask, "l" => l, "ϵb"=>ϵb, "slope"=>slope, "delta1"=>delta1, "delta2"=>delta2,"rho_ml"=>rho_ml,"drho"=>drho,"rho0"=>rho0, "c0"=>c0, "c1"=>c1,"z0"=>z0, "z1"=>z1,"Lz"=>Lz,"Lx"=>Lx,"Ly"=>Ly,"x0"=>x0)
)

The current version of Oceananigans that I'm running is:

println(Oceananigans.VERSION)
1.11.5

For Julia, it is the same 1.11.5

Any ideas on what I could change or on what might be causing this issue would be greatly appreciated. Thank you in advance for your help!

[glwagner]

I think this may be overlapping issue with #4036

[rafacmartins]

Oh, my bad! Should I delete this discussion and just add it as a comment on #4397?
Also, I just noticed that my Oceananigans.VERSION code was incorrect — it prints the Julia version, not the Oceananigans one.
My actual Oceananigans version is the latest one (0.96.21).

[glwagner]

No worries at all! Yes I think it's best to coalesce the discussion on the other issue