use the dealiased value for non-conservative convection

Ceyron · Oct 22, 2024 · c6daa20 · c6daa20
1 parent 6fc5697
commit c6daa20
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/exponax/nonlin_fun/_convection.py b/exponax/nonlin_fun/_convection.py
@@ -164,7 +164,9 @@ def _multi_channel_nonconservative_eval(
             )
         u_hat_dealiased = self.dealias(u_hat)
         u = self.ifft(u_hat_dealiased)
-        nabla_u = self.ifft(self.derivative_operator[None, :] * u_hat[:, None])
+        nabla_u = self.ifft(
+            self.derivative_operator[None, :] * u_hat_dealiased[:, None]
+        )
         conv_u = jnp.sum(
             u[None, :] * nabla_u,
             axis=1,
@@ -225,7 +227,7 @@ def _single_channel_nonconservative_eval(
         """
         u_hat_dealiased = self.dealias(u_hat)
         u = self.ifft(u_hat_dealiased)
-        nabla_u = self.ifft(self.derivative_operator * u_hat)
+        nabla_u = self.ifft(self.derivative_operator * u_hat_dealiased)
         conv_u = jnp.sum(
             u * nabla_u,
             axis=0,