Refactor allocation monitoring

perf/allocs.jl

@@ -1,17 +1,45 @@
 import Thermodynamics
 import RootSolvers
 import ClimaParams
-import ReportMetrics
+import Profile, ProfileCanvas
 
-dirs_to_monitor =
-    [".", pkgdir(Thermodynamics), pkgdir(RootSolvers), pkgdir(ClimaParams)]
+include("common.jl")
 
-for constructor in ["ρeq", "pθq", "pTq"]
-    ENV["ALLOCATION_CONSTRUCTOR"] = constructor
-    ReportMetrics.report_allocs(;
-        job_name = constructor,
-        run_cmd = `$(Base.julia_cmd()) --project=perf/ --track-allocation=all perf/alloc_per_constructor.jl`,
-        dirs_to_monitor = dirs_to_monitor,
-        process_filename = x -> last(split(x, "packages/")),
-    )
-end
+thermo_state_ρeq(param_set, ρ, e_int, q_tot)
+Profile.Allocs.clear()
+Profile.Allocs.@profile sample_rate = sampling_rate thermo_state_ρeq(
+    param_set,
+    ρ,
+    e_int,
+    q_tot,
+)
+results = Profile.Allocs.fetch()
+Profile.Allocs.clear()
+profile = ProfileCanvas.view_allocs(results)
+ProfileCanvas.html_file(joinpath(output_dir, "allocs_ρeq.html"), profile)
+
+thermo_state_pθq(param_set, p, θ_liq_ice, q_tot)
+Profile.Allocs.clear()
+Profile.Allocs.@profile sample_rate = sampling_rate thermo_state_pθq(
+    param_set,
+    p,
+    θ_liq_ice,
+    q_tot,
+)
+results = Profile.Allocs.fetch()
+Profile.Allocs.clear()
+profile = ProfileCanvas.view_allocs(results)
+ProfileCanvas.html_file(joinpath(output_dir, "allocs_pθq.html"), profile)
+
+thermo_state_pTq(param_set, p, T, q_tot)
+Profile.Allocs.clear()
+Profile.Allocs.@profile sample_rate = sampling_rate thermo_state_pTq(
+    param_set,
+    p,
+    T,
+    q_tot,
+)
+results = Profile.Allocs.fetch()
+Profile.Allocs.clear()
+profile = ProfileCanvas.view_allocs(results)
+ProfileCanvas.html_file(joinpath(output_dir, "allocs_pTq.html"), profile)

perf/benchmark.jl

@@ -2,11 +2,22 @@ include("common.jl")
 
 @testset "Thermodynamics - Performance pθq constructor" begin
 
-    trial = BenchmarkTools.@benchmark $thermo_state_ρeq()
+    trial = BenchmarkTools.@benchmark thermo_state_ρeq(
+        $param_set,
+        $ρ,
+        $e_int,
+        $q_tot,
+    )
     show(stdout, MIME("text/plain"), trial)
-    trial = BenchmarkTools.@benchmark $thermo_state_pθq()
+    trial = BenchmarkTools.@benchmark thermo_state_pθq(
+        $param_set,
+        $p,
+        $θ_liq_ice,
+        $q_tot,
+    )
     show(stdout, MIME("text/plain"), trial)
-    trial = BenchmarkTools.@benchmark $thermo_state_pTq()
+    trial =
+        BenchmarkTools.@benchmark thermo_state_pTq($param_set, $p, $T, $q_tot)
     show(stdout, MIME("text/plain"), trial)
 
 end

perf/common.jl

@@ -10,18 +10,27 @@ import ClimaParams as CP
 const FT = Float64
 const param_set = TP.ThermodynamicsParameters(FT)
 
-ArrayType = Array{FT}
-profiles = TD.TestedProfiles.PhaseEquilProfiles(param_set, ArrayType)
-(; e_int, T, ρ, p, θ_liq_ice, q_tot) = profiles
-
-function thermo_state_ρeq()
-    ts = TD.PhaseEquil_ρeq.(param_set, ρ, e_int, q_tot)
+function thermo_state_ρeq(param_set, _ρ, _e_int, _q_tot)
+    for (ρ, e_int, q_tot) in zip(_ρ, _e_int, _q_tot)
+        ts = TD.PhaseEquil_ρeq(param_set, ρ, e_int, q_tot)
+    end
+    return nothing
 end
 
-function thermo_state_pθq()
-    ts = TD.PhaseEquil_pθq.(param_set, p, θ_liq_ice, q_tot)
+function thermo_state_pθq(param_set, _p, _θ_liq_ice, _q_tot)
+    for (_p, _θ_liq_ice, _q_tot) in zip(p, θ_liq_ice, q_tot)
+        ts = TD.PhaseEquil_pθq(param_set, p, θ_liq_ice, q_tot)
+    end
+    return nothing
 end
 
-function thermo_state_pTq()
-    ts = TD.PhaseEquil_pTq.(param_set, p, T, q_tot)
+function thermo_state_pTq(param_set, _p, _T, _q_tot)
+    for (_p, _T, _q_tot) in zip(p, T, q_tot)
+        ts = TD.PhaseEquil_pTq(param_set, p, T, q_tot)
+    end
+    return nothing
 end
+
+ArrayType = Array{FT}
+profiles = TD.TestedProfiles.PhaseEquilProfiles(param_set, ArrayType)
+(; e_int, T, ρ, p, θ_liq_ice, q_tot) = profiles