From ee86b4863f39fba3682c859fe141dd7338b73a4c Mon Sep 17 00:00:00 2001 From: Julia Sloan Date: Mon, 8 Apr 2024 15:09:00 -0700 Subject: [PATCH] set device CPUMultiThreaded --- .buildkite/benchmarks/pipeline.yml | 2 ++ experiments/AMIP/coupler_driver.jl | 1 + 2 files changed, 3 insertions(+) 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)