diff --git a/.buildkite/benchmarks/pipeline.yml b/.buildkite/benchmarks/pipeline.yml index a90186c9b2..67f2d7462e 100644 --- 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 index b479c87574..71106251f5 100644 --- 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)