fix vertical_transport with vanleer_limiter on GPU

nefrathenrici · nefrathenrici · commit 0189827ceee2 · 2025-11-13T15:08:12.000-08:00
diff --git a/src/prognostic_equations/implicit/implicit_tendency.jl b/src/prognostic_equations/implicit/implicit_tendency.jl
@@ -86,6 +86,8 @@ end
     function vertical_transport(ᶜρ, ᶠu³, ᶜχ, dt, ::Val{:vanleer_limiter})
         ᶜJ = Fields.local_geometry_field(axes(ᶜρ)).J
         ᶠJ = Fields.local_geometry_field(axes(ᶠu³)).J
+        FT = eltype(ᶠJ)
+        dt = FT(dt)
         return @. lazy(
             -(ᶜadvdivᵥ(ᶠinterp(ᶜρ * ᶜJ) / ᶠJ * ᶠlin_vanleer(ᶠu³, ᶜχ, dt))),
         )

Original file line number	Diff line number	Diff line change
`@@ -86,6 +86,8 @@ end`
`86`	`86`	`function vertical_transport(ᶜρ, ᶠu³, ᶜχ, dt, ::Val{:vanleer_limiter})`
`87`	`87`	`ᶜJ = Fields.local_geometry_field(axes(ᶜρ)).J`
`88`	`88`	`ᶠJ = Fields.local_geometry_field(axes(ᶠu³)).J`
	`89`	`+ FT = eltype(ᶠJ)`
	`90`	`+ dt = FT(dt)`
`89`	`91`	`return @. lazy(`
`90`	`92`	`-(ᶜadvdivᵥ(ᶠinterp(ᶜρ * ᶜJ) / ᶠJ * ᶠlin_vanleer(ᶠu³, ᶜχ, dt))),`
`91`	`93`	`)`