From 0e1ccf1a29850f9444a1929de517310d9728cff2 Mon Sep 17 00:00:00 2001 From: Charles Kawczynski Date: Wed, 14 Feb 2024 14:01:31 -0500 Subject: [PATCH] Try to fix inference --- src/closures/entr_detr.jl | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/src/closures/entr_detr.jl b/src/closures/entr_detr.jl index debead392..fa5ac729c 100644 --- a/src/closures/entr_detr.jl +++ b/src/closures/entr_detr.jl @@ -335,7 +335,7 @@ function compute_phys_entr_detr!( Π = non_dimensional_groups(εδ_closure, εδ_model_vars) @assert length(Π) == n_Π_groups(edmf) for Π_i in 1:length(entrainment_Π_subset(edmf)) - Π_groups[Π_i][k] = Π[Π_i] + Π_groups.:($Π_i)[k] = Π.:($Π_i)::FT end # update fractional and turbulent entr/detr @@ -422,6 +422,7 @@ function compute_ml_entr_detr!( w_up = aux_up_f[i].w w_en = aux_en_f.w w_gm = prog_gm_f.w + Π_groups = aux_up[i].Π_groups @. w_up_c = Ic(w_up) @. w_en_c = Ic(w_en) @inbounds for k in real_center_indices(grid) @@ -455,7 +456,7 @@ function compute_ml_entr_detr!( Π = non_dimensional_groups(εδ_closure, εδ_model_vars) @assert length(Π) == n_Π_groups(edmf) for Π_i in 1:length(entrainment_Π_subset(edmf)) - aux_up[i].Π_groups[Π_i][k] = Π[Π_i] + Π_groups.:($Π_i)[k] = Π.:($Π_i)::FT end # update fractional and turbulent entr/detr # fractional, turbulent & nondimensional entrainment @@ -558,12 +559,12 @@ function compute_ml_entr_detr!( Π = non_dimensional_groups(εδ_model, εδ_model_vars) @assert length(Π) == n_Π_groups(edmf) for Π_i in 1:length(entrainment_Π_subset(edmf)) - aux_up[i].Π_groups[Π_i][k] = Π[Π_i] + Π_groups.:($Π_i)[k] = Π.:($Π_i)::FT end else for Π_i in 1:length(entrainment_Π_subset(edmf)) - aux_up[i].Π_groups[Π_i][k] = 0.0 + Π_groups.:($Π_i)[k] = FT(0) end end end