oops, forgot not everyone does time-parallelism again

firedrakeproject · May 1, 2024 · e76ffe6 · e76ffe6
1 parent afb235b
commit e76ffe6
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/gusto/io.py b/gusto/io.py
@@ -688,6 +688,10 @@ def dump(self, state_fields, t, step, initial_steps=None):
                 completed by a multi-level time scheme. Defaults to None.
         """
         output = self.output
+        if self.ensemble is not None:
+            write_file = self.ensemble.ensemble_comm.rank == 0
+        else:
+            write_file = True
 
         # Diagnostics:
         # Compute diagnostic fields
@@ -726,7 +730,7 @@ def dump(self, state_fields, t, step, initial_steps=None):
                 # dump fields
                 self.write_nc_dump(t)
 
-            if output.dump_vtus and self.ensemble.ensemble_comm.rank == 0:
+            if output.dump_vtus and write_file:
                 # dump fields
                 self.pvd_dumpfile.write(*self.to_dump)