Update moment_kinetics/src/em_fields.jl

Avoid allocating an intermediate array for -Epar. Co-authored-by: John Omotani <[email protected]>
mabarnes · Jan 13, 2025 · c0f9982 · c0f9982
1 parent 25ad9f2
commit c0f9982
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/moment_kinetics/src/em_fields.jl b/moment_kinetics/src/em_fields.jl
@@ -201,7 +201,8 @@ function calculate_phi_from_Epar!(phi, Epar, r, z, z_spectral)
         # calculate phi on each local rank, up to a constant
         # phi[1,:] = 0.0 by convention here
         @loop_r ir begin
-            @views indefinite_integral!(phi[:,ir], -Epar[:,ir], z, z_spectral)
+            @views @. z.scratch = -Epar[:,ir]
+            @views indefinite_integral!(phi[:,ir], z.scratch, z, z_spectral)
         end
 
         # Restore the constant offset from the lower boundary