add interior in chmy functions

Author: Iddingsite

ext/ChmyExt1D.jl

@@ -93,17 +93,17 @@ function WENO_step!(u::T_field, v::NamedTuple{(:x,), <:Tuple{<:T_field}}, weno::
     launch(arch, grid, WENO_flux_chmy_1D => (fl.x, fr.x, u, boundary, nx, χ, γ, ζ, ϵ, grid))
     launch(arch, grid, WENO_semi_discretisation_weno5_chmy_1D! => (du, fl, fr, v, stag, Δx_, grid))
 
-    ut .= u .- Δt .* du
+    interior(ut) .= interior(u) .- Δt .* interior(du)
 
     launch(arch, grid, WENO_flux_chmy_1D => (fl.x, fr.x, ut, boundary, nx, χ, γ, ζ, ϵ, grid))
     launch(arch, grid, WENO_semi_discretisation_weno5_chmy_1D! => (du, fl, fr, v, stag, Δx_, grid))
 
-    ut .= 0.75 .* u .+ 0.25 .* ut .- 0.25 .* Δt .* du
+    interior(ut) .= 0.75 .* interior(u) .+ 0.25 .* interior(ut) .- 0.25 .* Δt .* interior(du)
 
     launch(arch, grid, WENO_flux_chmy_1D => (fl.x, fr.x, ut, boundary, nx, χ, γ, ζ, ϵ, grid))
     launch(arch, grid, WENO_semi_discretisation_weno5_chmy_1D! => (du, fl, fr, v, stag, Δx_, grid))
 
-    u .= inv(3.0) .* u .+ 2.0 / 3.0 .* ut .- 2.0 / 3.0 .* Δt .* du
+    interior(u) .= inv(3.0) .* interior(u) .+ 2.0 / 3.0 .* interior(ut) .- 2.0 / 3.0 .* Δt .* interior(du)
 
     return synchronize(backend)
 end

ext/ChmyExt2D.jl

@@ -163,19 +163,19 @@ function WENO_step!(u::T_field, v, weno::FiniteDiffWENO5.WENOScheme, Δt, Δx, 
     launch(arch, grid, WENO_flux_chmy_2D_y => (fl.y, fr.y, u, boundary, ny, χ, γ, ζ, ϵ, grid))
     launch(arch, grid, WENO_semi_discretisation_weno5_chmy_2D! => (du, fl, fr, v, stag, Δx_, Δy_, grid))
 
-    ut .= @muladd u .- Δt .* du
+    interior(ut) .= @muladd interior(u) .- Δt .* interior(du)
 
     launch(arch, grid, WENO_flux_chmy_2D_x => (fl.x, fr.x, ut, boundary, nx, χ, γ, ζ, ϵ, grid))
     launch(arch, grid, WENO_flux_chmy_2D_y => (fl.y, fr.y, ut, boundary, ny, χ, γ, ζ, ϵ, grid))
     launch(arch, grid, WENO_semi_discretisation_weno5_chmy_2D! => (du, fl, fr, v, stag, Δx_, Δy_, grid))
 
-    ut .= @muladd 0.75 .* u .+ 0.25 .* ut .- 0.25 .* Δt .* du
+    interior(ut) .= @muladd 0.75 .* interior(u) .+ 0.25 .* interior(ut) .- 0.25 .* Δt .* interior(du)
 
     launch(arch, grid, WENO_flux_chmy_2D_x => (fl.x, fr.x, ut, boundary, nx, χ, γ, ζ, ϵ, grid))
     launch(arch, grid, WENO_flux_chmy_2D_y => (fl.y, fr.y, ut, boundary, ny, χ, γ, ζ, ϵ, grid))
     launch(arch, grid, WENO_semi_discretisation_weno5_chmy_2D! => (du, fl, fr, v, stag, Δx_, Δy_, grid))
 
-    u .= @muladd inv(3.0) .* u .+ 2.0 / 3.0 .* ut .- 2.0 / 3.0 .* Δt .* du
+    interior(u) .= @muladd inv(3.0) .* interior(u) .+ 2.0 / 3.0 .* interior(ut) .- 2.0 / 3.0 .* Δt .* interior(du)
 
     return synchronize(backend)
 end

ext/ChmyExt3D.jl

@@ -220,21 +220,21 @@ function WENO_step!(u::T_field, v, weno::FiniteDiffWENO5.WENOScheme, Δt, Δx, 
     launch(arch, grid, WENO_flux_chmy_3D_z => (fl.z, fr.z, u, boundary, nz, χ, γ, ζ, ϵ, grid))
     launch(arch, grid, WENO_semi_discretisation_weno5_chmy_3D! => (du, fl, fr, v, stag, Δx_, Δy_, Δz_, grid))
 
-    ut .= @muladd u .- Δt .* du
+    interior(ut) .= @muladd interior(u) .- Δt .* interior(du)
 
     launch(arch, grid, WENO_flux_chmy_3D_x => (fl.x, fr.x, ut, boundary, nx, χ, γ, ζ, ϵ, grid))
     launch(arch, grid, WENO_flux_chmy_3D_y => (fl.y, fr.y, ut, boundary, ny, χ, γ, ζ, ϵ, grid))
     launch(arch, grid, WENO_flux_chmy_3D_z => (fl.z, fr.z, ut, boundary, nz, χ, γ, ζ, ϵ, grid))
     launch(arch, grid, WENO_semi_discretisation_weno5_chmy_3D! => (du, fl, fr, v, stag, Δx_, Δy_, Δz_, grid))
 
-    ut .= @muladd 0.75 .* u .+ 0.25 .* ut .- 0.25 .* Δt .* du
+    interior(ut) .= @muladd 0.75 .* interior(u) .+ 0.25 .* interior(ut) .- 0.25 .* Δt .* interior(du)
 
     launch(arch, grid, WENO_flux_chmy_3D_x => (fl.x, fr.x, ut, boundary, nx, χ, γ, ζ, ϵ, grid))
     launch(arch, grid, WENO_flux_chmy_3D_y => (fl.y, fr.y, ut, boundary, ny, χ, γ, ζ, ϵ, grid))
     launch(arch, grid, WENO_flux_chmy_3D_z => (fl.z, fr.z, ut, boundary, nz, χ, γ, ζ, ϵ, grid))
     launch(arch, grid, WENO_semi_discretisation_weno5_chmy_3D! => (du, fl, fr, v, stag, Δx_, Δy_, Δz_, grid))
 
-    u .= @muladd inv(3.0) .* u .+ 2.0 / 3.0 .* ut .- 2.0 / 3.0 .* Δt .* du
+    interior(u) .= @muladd inv(3.0) .* interior(u) .+ 2.0 / 3.0 .* interior(ut) .- 2.0 / 3.0 .* Δt .* interior(du)
 
     return synchronize(backend)
 end