Field type inconsistency with Reactant

[dkytezab]

using Reactant
using CUDA
using Oceananigans
using Oceananigans.Architectures: ReactantState
using Oceananigans.Fields: CenterField, compute!

grid = RectilinearGrid(
    ReactantState();
    size = (4, 4, 4), halo = (3, 3, 3), extent = (1, 1, 1),
    topology = (Periodic, Periodic, Bounded)
)

field = CenterField(grid)

eltypes(f) = @show eltype(f), eltype(CenterField(f.grid))

operate(f) = compute!(Field(f + CenterField(f.grid)))

@compile eltypes(field)
@compile raise = true operate(field)

(eltype(f), eltype(CenterField(f.grid))) = (Float64, Reactant.TracedRNumber{Float64})
ERROR: LoadError: InvalidIRError: compiling MethodInstance for Oceananigans.AbstractOperations.gpu__compute!(::KernelAbstractions.CompilerMetadata{…}, ::OffsetArrays.OffsetArray{…}, ::Oceananigans.AbstractOperations.BinaryOperation{…}) resulted in invalid LLVM IR
Reason: unsupported dynamic function invocation (call to getindex)
Stacktrace:
 [1] gpu__compute!
   @ ~/.julia/packages/KernelAbstractions/scVtc/src/macros.jl:332
 [2] gpu__compute!
   @ ./none:0
Hint: catch this exception as `err` and call `code_typed(err; interactive = true)` to introspect the erroneous code with Cthulhu.jl
Stacktrace
...

cc @Pangoraw

[dkytezab]

One fix could be to replace T with T = Grids.unwrapped_eltype(eltype(data)).

Oceananigans.jl/src/Fields/field.jl

Lines 30 to 34 in bb606d5

	function Field{LX, LY, LZ}(grid::G, data::D, bcs::B, indices::I, op::O, status::S, buffers::F) where {LX, LY, LZ, G, D, B, O, S, I, F}
	T = eltype(data)
	@apply_regionally local_bcs = construct_boundary_conditions_kernels(bcs, data, grid, (LX(), LY(), LZ()), indices) # Adding the kernels to the bcs
	return new{LX, LY, LZ, O, G, I, D, T, typeof(local_bcs), S, F}(grid, data, local_bcs, indices, op, status, buffers)
	end