get device suffix correctly

juliasloan25 · juliasloan25 · commit 58a18338afbd · 2024-03-21T17:01:28.000-07:00
diff --git a/experiments/AMIP/coupler_driver.jl b/experiments/AMIP/coupler_driver.jl
@@ -762,7 +762,7 @@ debug(cs, debug_dir * "/8_after_solve_")
 
 
 # Save states after simulation for CPU/GPU comparison
-device_suffix = comms_ctx isa ClimaComms.AbstractCPUDevice ? "cpu" : "gpu"
+device_suffix = comms_ctx.device isa ClimaComms.AbstractCPUDevice ? "cpu" : "gpu"
 
 # Extract atmos state variables
 atmos_ρe_tot = cs.model_sims.atmos_sim.integrator.u.c.ρe_tot
