horizontal_space netcdf writer

CliMA · May 15, 2024 · b85e136 · b85e136
1 parent 43ca4f1
commit b85e136
Showing 1 changed file with 12 additions and 3 deletions.
diff --git a/src/netcdf_writer.jl b/src/netcdf_writer.jl
@@ -148,12 +148,21 @@ function NetCDFWriter(
         hpts,
     )
     zcoords = Geometry.ZPoint.(vpts)
-
     remapper = Remapper(space, hcoords, zcoords)
+    comms_ctx = ClimaComms.context(space)
 
-    coords_z = Fields.coordinate_field(space).z
+    if is_horizontal_space
+        interpolated_physical_z = []
+    else
+        coords_z = Fields.coordinate_field(space).z
+        maybe_move_to_cpu =
+            ClimaComms.device(coords_z) isa ClimaComms.CUDADevice &&
+            ClimaComms.iamroot(comms_ctx) ? Array : identity
+
+        interpolated_physical_z =
+            maybe_move_to_cpu(interpolate(remapper, coords_z))
+    end
 
-    comms_ctx = ClimaComms.context(space)
     maybe_move_to_cpu =
         ClimaComms.device(coords_z) isa ClimaComms.CUDADevice &&
         ClimaComms.iamroot(comms_ctx) ? Array : identity