Avoid gpu-incompatible remapper logic

Author: charleskawczynski

src/Remapping/Remapping.jl

@@ -16,6 +16,7 @@ import ..DataLayouts,
     ..Hypsography
 import ClimaCore.Utilities: half
 import ClimaCore.Spaces: cuda_synchronize
+import ..to_cpu
 
 using ..RecursiveApply
 

src/Remapping/distributed_remapping.jl

@@ -399,10 +399,15 @@ function _Remapper(
         )
         num_dims = num_hdims
     else
+        cpu_space = if ClimaComms.device(space) isa ClimaComms.AbstractCPUDevice
+            space
+        else
+            to_cpu(space)
+        end
         vert_interpolation_weights =
-            ArrayType(vertical_interpolation_weights(space, target_zcoords))
+            ArrayType(vertical_interpolation_weights(cpu_space, target_zcoords))
         vert_bounding_indices =
-            ArrayType(vertical_bounding_indices(space, target_zcoords))
+            ArrayType(vertical_bounding_indices(cpu_space, target_zcoords))
 
         # We have to add one extra dimension with respect to the bitmask/local_horiz_indices
         # because we are going to store the values for the columns
@@ -463,10 +468,16 @@ function _Remapper(
     FT = Spaces.undertype(space)
     ArrayType = ClimaComms.array_type(space)
 
+    cpu_space = if ClimaComms.device(space) isa ClimaComms.AbstractCPUDevice
+        space
+    else
+        to_cpu(space)
+    end
+
     vert_interpolation_weights =
-        ArrayType(vertical_interpolation_weights(space, target_zcoords))
+        ArrayType(vertical_interpolation_weights(cpu_space, target_zcoords))
     vert_bounding_indices =
-        ArrayType(vertical_bounding_indices(space, target_zcoords))
+        ArrayType(vertical_bounding_indices(cpu_space, target_zcoords))
 
     local_interpolated_values =
         ArrayType(zeros(FT, (length(target_zcoords), buffer_length)))