Use one-liner instead of defining geofac

Avoids extra allocation.
mabarnes · Mar 13, 2024 · 655b951 · 655b951
1 parent 7ddf1a2
commit 655b951
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/moment_kinetics/src/z_advection.jl b/moment_kinetics/src/z_advection.jl
@@ -106,8 +106,7 @@ function update_speed_z!(advect, upar, vth, evolve_upar, evolve_ppar, fields, vp
                 # vpa bzed
                 @. @views advect.speed[:,ivpa,ivperp,ir] = vpa.grid[ivpa]*bzed[:,ir]
                 # ExB drift
-                @. geofac = bzeta[:,ir]*jacobian[:,ir]/Bmag[:,ir]
-                @. @views advect.speed[:,ivpa,ivperp,ir] += ExBfac*geofac*fields.Er[:,ir]
+                @. @views advect.speed[:,ivpa,ivperp,ir] += ExBfac*bzeta[:,ir]*jacobian[:,ir]/Bmag[:,ir]*fields.Er[:,ir]
                 # magnetic curvature drift
                 @. @views advect.speed[:,ivpa,ivperp,ir] += rhostar*(vpa.grid[ivpa]^2)*cvdriftz[:,ir]
                 # magnetic grad B drift