Try debugging

CliMA · Jan 24, 2025 · 24e8f62 · 24e8f62
1 parent af35525
commit 24e8f62
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/.buildkite/pipeline.yml b/.buildkite/pipeline.yml
@@ -342,13 +342,13 @@ steps:
 
       - label: "Unit: distributed remapping (1 process)"
         key: distributed_remapping_1proc
-        command: "julia --color=yes --check-bounds=yes --project=.buildkite test/Remapping/distributed_remapping.jl"
+        command: "julia --color=yes --check-bounds=yes -g2 --project=.buildkite test/Remapping/distributed_remapping.jl"
         env:
           CLIMACOMMS_DEVICE: "CPU"
 
       - label: "Unit: distributed remapping (2 processes)"
         key: distributed_remapping_2procs
-        command: "srun julia --color=yes --check-bounds=yes --project=.buildkite test/Remapping/distributed_remapping.jl"
+        command: "srun julia --color=yes --check-bounds=yes -g2 --project=.buildkite test/Remapping/distributed_remapping.jl"
         env:
           CLIMACOMMS_CONTEXT: "MPI"
           CLIMACOMMS_DEVICE: "CPU"

diff --git a/ext/cuda/remapping_distributed.jl b/ext/cuda/remapping_distributed.jl
@@ -80,14 +80,15 @@ function set_interpolated_values_kernel!(
         h = local_horiz_indices[i]
         v_lo, v_hi = vert_bounding_indices[j]
         A, B = vert_interpolation_weights[j]
+        fvals = field_values[k]
         out[i, j, k] = 0
         for t in 1:Nq, s in 1:Nq
             out[i, j, k] +=
                 I1[i, t] *
                 I2[i, s] *
                 (
-                    A * field_values[k][CI(t, s, 1, v_lo, h)] +
-                    B * field_values[k][CI(t, s, 1, v_hi, h)]
+                    A * fvals[CI(t, s, 1, v_lo, h)] +
+                    B * fvals[CI(t, s, 1, v_hi, h)]
                 )
         end
     end