Allow interpolation on Distributed architectures

src/Fields/interpolate.jl

@@ -7,6 +7,8 @@ using Oceananigans.Grids: topology, node,
                           ZRegOrthogonalSphericalShellGrid,
                           RectilinearGrid, LatitudeLongitudeGrid
 
+using Oceananigans.Architectures: child_architecture
+
 # GPU-compatile middle point calculation
 @inline middle_point(l, h) = Base.unsafe_trunc(Int, (l + h) / 2)
 
@@ -313,6 +315,12 @@ function interpolate!(to_field::Field, from_field::AbstractField)
 
     to_arch   = architecture(to_field)
     from_arch = architecture(from_field)
+
+    # In case architectures are `Distributed` we 
+    # verify that the fields are on the same child architecture
+    to_arch   = child_architecture(to_arch)
+    from_arch = child_architecture(from_arch)
+
     if !isnothing(from_arch) && to_arch != from_arch
         msg = "Cannot interpolate! because from_field is on $from_arch while to_field is on $to_arch."
         throw(ArgumentError(msg))

test/test_distributed_hydrostatic_model.jl

@@ -79,7 +79,7 @@ Ny = 32
 for arch in archs
     @testset "Testing distributed solid body rotation" begin
         grid = LatitudeLongitudeGrid(arch, size = (Nx, Ny, 1),
-                                     halo = (3, 3, 3),
+                                     halo = (4, 4, 4),
                                      latitude = (-80, 80),
                                      longitude = (-160, 160),
                                      z = (-1, 0),