Ensure normalization works for spheres

DedalusProject · Feb 6, 2024 · 4f96bb9 · 4f96bb9
1 parent b0ee129
commit 4f96bb9
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/dedalus/core/field.py b/dedalus/core/field.py
@@ -800,9 +800,10 @@ def allreduce_L2_norm(self, normalize_volume=True):
         else:
             raise ValueError("Norms only implemented up to rank-2 tensors.")
         # Compute L2 norm
-        norm_sq = operators.Integrate(self_inner_product).evaluate().allreduce_data_max()
         if normalize_volume:
-            norm_sq /= self.domain.volume
+            norm_sq = operators.Average(self_inner_product).evaluate().allreduce_data_max(layout='g')
+        else:
+            norm_sq = operators.Integrate(self_inner_product).evaluate().allreduce_data_max(layout='g')
         return norm_sq ** 0.5
 
     def normalize(self, normalize_volume=True):