From 4a6c201a56c74da35c7229f921c8879a66f5c567 Mon Sep 17 00:00:00 2001
From: Zhaoyi Shen <11598433+szy21@users.noreply.github.com>
Date: Wed, 10 Jul 2024 15:30:55 -0700
Subject: [PATCH] use 1 gpu for amip pipeline

---
 .buildkite/amip/pipeline.yml | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/.buildkite/amip/pipeline.yml b/.buildkite/amip/pipeline.yml
index 54ecd6b97..0eacbf0bc 100644
--- a/.buildkite/amip/pipeline.yml
+++ b/.buildkite/amip/pipeline.yml
@@ -50,12 +50,11 @@ steps:
         artifact_paths: "experiments/ClimaEarth/output/amip/amip_artifacts/*"
         timeout_in_minutes: 4320
         env:
-          CLIMACOMMS_CONTEXT: "MPI"
           CLIMACOMMS_DEVICE: "CUDA"
         agents:
           slurm_gpus_per_task: 1
           slurm_cpus_per_task: 4
-          slurm_ntasks: 4
+          slurm_ntasks: 1
           slurm_mem: 30GB
         soft_fail: true