clean

CliMA · May 2, 2024 · 4fa6660 · 4fa6660
1 parent 69d38a3
commit 4fa6660
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/experiments/AMIP/coupler_driver.jl b/experiments/AMIP/coupler_driver.jl
@@ -896,7 +896,9 @@ if ClimaComms.iamroot(comms_ctx)
     end
 
     ## plot all model states and coupler fields (useful for debugging) TODO: make MPI & GPU compatible
-    comms_ctx isa ClimaComms.SingletonCommsContext && debug(cs, joinpath(COUPLER_ARTIFACTS_DIR, "end_"))
+    comms_ctx.device == ClimaComms.CPUSingleThreaded() &&
+        comms_ctx isa ClimaComms.SingletonCommsContext &&
+        debug(cs, joinpath(COUPLER_ARTIFACTS_DIR, "end_"))
 
     if isinteractive()
         ## clean up for interactive runs, retain all output otherwise