set device CPUMultiThreaded

CliMA · Apr 9, 2024 · ee86b48 · ee86b48
1 parent f9c3017
commit ee86b48
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/.buildkite/benchmarks/pipeline.yml b/.buildkite/benchmarks/pipeline.yml
@@ -55,6 +55,7 @@ steps:
         artifact_paths: "experiments/AMIP/output/aquaplanet_diagedmf/*"
         env:
           BUILD_HISTORY_HANDLE: ""
+          CLIMACOMMS_DEVICE: "CPUMultiThreaded"
         agents:
           slurm_ntasks_per_node: 16
           slurm_nodes: 1
@@ -66,6 +67,7 @@ steps:
         artifact_paths: "experiments/AMIP/output/amip/amip_diagedmf_artifacts/*"
         env:
           BUILD_HISTORY_HANDLE: ""
+          CLIMACOMMS_DEVICE: "CPUMultiThreaded"
         agents:
           slurm_ntasks_per_node: 16
           slurm_nodes: 1

diff --git a/experiments/AMIP/coupler_driver.jl b/experiments/AMIP/coupler_driver.jl
@@ -187,6 +187,7 @@ then `ClimaComms` automatically selects the device from which this code is calle
 =#
 
 using ClimaComms
+
 comms_ctx = get_comms_context(parsed_args)
 ClimaComms.init(comms_ctx)
-Original file line number
+Diff line change
@@ Expand Up @@
     =#
     using ClimaComms
     comms_ctx = get_comms_context(parsed_args)
     ClimaComms.init(comms_ctx)
@@ Expand Down @@