cuda

CliMA · Nov 23, 2023 · ec0e27d · ec0e27d
1 parent fd1f5ca
commit ec0e27d
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/.buildkite/pipeline.yml b/.buildkite/pipeline.yml
@@ -92,6 +92,13 @@ steps:
         env:
           CLIMACOMMS_CONTEXT: "MPI"
 
+      - label: "reproduce CUDA"
+        command: "srun julia --color=yes --project=examples -e 'import CUDA; import ClimaComms; ctx = ClimaComms.context(); ClimaComms.init(ctx); println(ctx)'"
+        agents:
+          slurm_ntasks: 2
+          slurm_mem: 16G
+        env:
+          CLIMACOMMS_CONTEXT: "MPI"
 
       - label: "ClimaAtmos config"
         command: "srun julia --color=yes --project=examples -e 'import ClimaAtmos; println(\"HELLO\");'"