convert fields GPU to CPU for plots

Author: juliasloan25

experiments/AMIP/Manifest.toml

@@ -2,7 +2,7 @@
 
 julia_version = "1.10.2"
 manifest_format = "2.0"
-project_hash = "284db48dcd39d58ad8518daa0c1caded1552fa4a"
+project_hash = "a198c94037ccdef0790bbbb0c9a214fd17aa95f9"
 
 [[deps.ADTypes]]
 git-tree-sha1 = "016833eb52ba2d6bea9fcb50ca295980e728ee24"

experiments/AMIP/Project.toml

@@ -1,4 +1,5 @@
 [deps]
+Adapt = "79e6a3ab-5dfb-504d-930d-738a2a938a0e"
 ArgParse = "c7e460c6-2fb9-53a9-8c5b-16f535851c63"
 ArtifactWrappers = "a14bc488-3040-4b00-9dc1-f6467924858a"
 AtmosphericProfilesLibrary = "86bc3604-9858-485a-bdbe-831ec50de11d"

experiments/AMIP/user_io/debug_plots.jl

@@ -3,6 +3,7 @@ using ClimaCorePlots
 using Printf
 using ClimaCoupler.Interfacer: ComponentModelSimulation, SurfaceModelSimulation
 using ClimaCore
+using Adapt
 
 # plotting functions for the coupled simulation
 """
@@ -42,15 +43,21 @@ function debug(cs_fields::NamedTuple, dir, cs_fields_ref = nothing)
         :z0m_S,
     )
     all_plots = []
+    cpu_comms_ctx = ClimaComms.SingletonCommsContext()
+
+
     for field_name in field_names
         field = getproperty(cs_fields, field_name)
-        push!(all_plots, Plots.plot(field, title = string(field_name) * print_extrema(field)))
+        # Convert field from GPU to CPU if necessary
+        cpu_field = Adapt.adapt(Array, field)
+
+        push!(all_plots, Plots.plot(cpu_field, title = string(field_name) * print_extrema(cpu_field)))
         if (field_name == :T_S) && (@isdefined debug_csf0)
             push!(
                 all_plots,
                 Plots.plot(
-                    field .- debug_csf0.T_S,
-                    title = string(field_name) * "_anom" * print_extrema(field),
+                    cpu_field .- debug_csf0.T_S,
+                    title = string(field_name) * "_anom" * print_extrema(cpu_field),
                     color = :viridis,
                 ),
             )
@@ -64,11 +71,14 @@ function debug(cs_fields::NamedTuple, dir, cs_fields_ref = nothing)
         all_plots = []
         for field_name in field_names
             field = getproperty(cs_fields, field_name)
+            # Convert field from GPU to CPU if necessary
+            cpu_field = Adapt.adapt(Array, field)
+
             push!(
                 all_plots,
                 Plots.plot(
-                    field .- getproperty(cs_fields_ref, field_name),
-                    title = string(field_name) * print_extrema(field),
+                    cpu_field .- getproperty(cs_fields_ref, field_name),
+                    title = string(field_name) * print_extrema(cpu_field),
                 ),
             )
         end
@@ -89,7 +99,10 @@ function debug(sim::ComponentModelSimulation, dir)
     all_plots = []
     for field_name in field_names
         field = get_field(sim, Val(field_name))
-        push!(all_plots, Plots.plot(field, title = string(field_name) * print_extrema(field)))
+        # Convert field from GPU to CPU if necessary
+        cpu_field = Adapt.adapt(Array, field)
+
+        push!(all_plots, Plots.plot(cpu_field, title = string(field_name) * print_extrema(cpu_field)))
     end
     fig = Plots.plot(all_plots..., size = (1500, 800))
     Plots.png(joinpath(dir, "debug_$(name(sim))"))

test/Project.toml

@@ -1,4 +1,5 @@
 [deps]
+Adapt = "79e6a3ab-5dfb-504d-930d-738a2a938a0e"
 Aqua = "4c88cf16-eb10-579e-8560-4a9242c79595"
 ArtifactWrappers = "a14bc488-3040-4b00-9dc1-f6467924858a"
 CairoMakie = "13f3f980-e62b-5c42-98c6-ff1f3baf88f0"