Fix compare cpu/gpu tests

The tests were only comparing the albedo function job
CliMA · Jun 26, 2024 · a7193e4 · a7193e4
1 parent 9c2abcd
commit a7193e4
Show file tree

Hide file tree

Showing 5 changed files with 16 additions and 18 deletions.
diff --git a/.buildkite/pipeline.yml b/.buildkite/pipeline.yml
@@ -198,20 +198,12 @@ steps:
 
   - group: "CPU/GPU comparisons"
     steps:
-      - label: "Compare GPU bucket with CPU bucket (functional albedo)"
+      - label: "Compare GPU bucket with CPU bucket"
         command: "julia --color=yes --project=.buildkite experiments/standalone/Bucket/compare_gpu_cpu_output.jl"
         depends_on:
           - "global_bucket_function_cpu"
           - "global_bucket_function_gpu"
-
-      - label: "Compare GPU bucket with CPU bucket (static map albedo)"
-        command: "julia --color=yes --project=.buildkite experiments/standalone/Bucket/compare_gpu_cpu_output.jl"
-        depends_on:
           - "global_bucket_staticmap_cpu"
           - "global_bucket_staticmap_gpu"
-
-      - label: "Compare GPU bucket with CPU bucket (temporal map albedo)"
-        command: "julia --color=yes --project=.buildkite experiments/standalone/Bucket/compare_gpu_cpu_output.jl"
-        depends_on:
           - "global_bucket_temporalmap_cpu"
           - "global_bucket_temporalmap_gpu"
diff --git a/experiments/standalone/Bucket/compare_gpu_cpu_output.jl b/experiments/standalone/Bucket/compare_gpu_cpu_output.jl
@@ -2,9 +2,15 @@ using DelimitedFiles
 using Statistics
 import ClimaLand
 outdir = joinpath(pkgdir(ClimaLand), "experiments/standalone/Bucket/artifacts")
-cpu_state = readdlm(joinpath(outdir, "tf_state_cpu.txt"), ',')
-gpu_state = readdlm(joinpath(outdir, "tf_state_gpu.txt"), ',')
-@show abs(maximum(cpu_state .- gpu_state))
-@show abs(median(cpu_state .- gpu_state))
-@show abs(mean(cpu_state .- gpu_state))
-@assert isapprox(cpu_state, gpu_state)
+function check(job)
+    cpu_state = readdlm(joinpath(outdir, "tf_cpu_$job.txt"), ',')
+    gpu_state = readdlm(joinpath(outdir, "tf_gpu_$job.txt"), ',')
+    @show abs(maximum(cpu_state .- gpu_state))
+    @show abs(median(cpu_state .- gpu_state))
+    @show abs(mean(cpu_state .- gpu_state))
+    @assert isapprox(cpu_state, gpu_state)
+end
+
+check("function")
+check("static")
+check("temporal")
diff --git a/experiments/standalone/Bucket/global_bucket_function.jl b/experiments/standalone/Bucket/global_bucket_function.jl
@@ -226,6 +226,6 @@ T_sfc = Array(Remapping.interpolate(remapper, prob.p.bucket.T_sfc))
 device_suffix =
     typeof(ClimaComms.context().device) <: ClimaComms.CPUSingleThreaded ?
     "cpu" : "gpu"
-open(joinpath(outdir, "tf_state_$device_suffix.txt"), "w") do io
+open(joinpath(outdir, "tf_$(device_suffix)_function.txt"), "w") do io
     writedlm(io, hcat(T_sfc[:], W[:], Ws[:], σS[:]), ',')
 end;
diff --git a/experiments/standalone/Bucket/global_bucket_staticmap.jl b/experiments/standalone/Bucket/global_bucket_staticmap.jl
@@ -305,7 +305,7 @@ sw_forcing = [
 ];
 
 # save prognostic state to CSV - for comparison between GPU and CPU output
-open(joinpath(outdir, "tf_state_$device_suffix.txt"), "w") do io
+open(joinpath(outdir, "tf_state_$(device_suffix)_static.txt"), "w") do io
     writedlm(io, hcat(T_sfc[end][:], W[end][:], Ws[end][:], σS[end][:]), ',')
 end;
 # animation settings

diff --git a/experiments/standalone/Bucket/global_bucket_temporalmap.jl b/experiments/standalone/Bucket/global_bucket_temporalmap.jl
@@ -277,7 +277,7 @@ F_sfc = [
 ];
 
 # save prognostic state to CSV - for comparison between GPU and CPU output
-open(joinpath(outdir, "tf_state_$device_suffix.txt"), "w") do io
+open(joinpath(outdir, "tf_state_$(device_suffix)_temporal.txt"), "w") do io
     writedlm(io, hcat(T_sfc[end][:], W[end][:], Ws[end][:], σS[end][:]), ',')
 end;
 # animation settings