From 5a429ca89a33974f02ffe423b754b450a53d6e39 Mon Sep 17 00:00:00 2001 From: Sam Abbott Date: Fri, 4 Oct 2024 11:54:53 +0100 Subject: [PATCH] Remove cumsum broadcast optimisation --- EpiAware/src/EpiInfModels/ExpGrowthRate.jl | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/EpiAware/src/EpiInfModels/ExpGrowthRate.jl b/EpiAware/src/EpiInfModels/ExpGrowthRate.jl index 5b142db0f..ffb412b2f 100644 --- a/EpiAware/src/EpiInfModels/ExpGrowthRate.jl +++ b/EpiAware/src/EpiInfModels/ExpGrowthRate.jl @@ -116,5 +116,5 @@ I_t = generated_quantities(latent_inf, θ) " @model function EpiAwareBase.generate_latent_infs(epi_model::ExpGrowthRate, rt) init_incidence ~ epi_model.initialisation_prior - return xexpy.(1.0, accumulate(+, rt; init = init_incidence)) + return xexpy.(1.0, init_incidence .+ cumsum(rt)) end