Use LazyBroadcast for diagnostics

CliMA · Aug 14, 2024 · 80dd655 · 80dd655
1 parent 41b8467
commit 80dd655
Show file tree

Hide file tree

Showing 8 changed files with 410 additions and 869 deletions.
diff --git a/Project.toml b/Project.toml
@@ -25,6 +25,7 @@ Insolation = "e98cc03f-d57e-4e3c-b70c-8d51efe9e0d8"
 Interpolations = "a98d9a8b-a2ab-59e6-89dd-64a1c18fca59"
 IntervalSets = "8197267c-284f-5f27-9208-e0e47529a953"
 Krylov = "ba0b0d4f-ebba-5204-a429-3ac8c609bfb7"
+LazyBroadcast = "9dccce8e-a116-406d-9fcc-a88ed4f510c8"
 LinearAlgebra = "37e2e46d-f89d-539d-b4ee-838fcccc9c8e"
 Logging = "56ddb016-857b-54e1-b83d-db4d58db5568"
 NCDatasets = "85f8d34a-cbdd-5861-8df4-14fed0d494ab"

diff --git a/examples/Manifest.toml b/examples/Manifest.toml
diff --git a/perf/Manifest.toml b/perf/Manifest.toml
diff --git a/src/diagnostics/Diagnostics.jl b/src/diagnostics/Diagnostics.jl
@@ -48,6 +48,8 @@ include(joinpath("..", "utils", "abbreviations.jl"))
 import ClimaDiagnostics:
     DiagnosticVariable, ScheduledDiagnostic, average_pre_output_hook!
 
+import LazyBroadcast: @lazy
+
 import ClimaDiagnostics.DiagnosticVariables: descriptive_short_name
 
 import ClimaDiagnostics.Schedules: EveryStepSchedule, EveryDtSchedule