fixup device_data commit c3fe045

ecmwf · Oct 14, 2024 · d383cac · d383cac
1 parent e00c86a
commit d383cac
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/atlas/array/ArraySpec.cc b/src/atlas/array/ArraySpec.cc
@@ -57,7 +57,7 @@ ArraySpec::ArraySpec(const ArrayShape& shape, const ArrayAlignment& alignment):
         allocated_size_ *= size_t(aligned_shape[j]);
         if( j < rank_ - 1) {
             // Assume contiguous device data!
-            device_strides_[j] = strides_[j+1] * shape[j];
+            device_strides_[j] = strides_[j+1] * shape[j+1];
         }
     }
     ATLAS_ASSERT(allocated_size_ == compute_aligned_size(size_t(shape_[0]) * size_t(strides_[0]), size_t(alignment)));
@@ -96,7 +96,7 @@ ArraySpec::ArraySpec(const ArrayShape& shape, const ArrayStrides& strides, const
         size_ *= size_t(shape_[j]);
         if( j < rank_ - 1) {
             // Assume contiguous device data!
-            device_strides_[j] = device_strides_[j+1] * shape[j];
+            device_strides_[j] = device_strides_[j+1] * shape[j+1];
         }
     }
     allocated_size_ = compute_aligned_size(size_t(shape_[0]) * size_t(strides_[0]), size_t(alignment));
@@ -146,7 +146,7 @@ ArraySpec::ArraySpec(const ArrayShape& shape, const ArrayStrides& strides, const
         }
         if( j < rank_ - 1) {
             // Assume contiguous device data!
-            device_strides_[j] = device_strides_[j+1] * shape[j];
+            device_strides_[j] = device_strides_[j+1] * shape[j+1];
         }
     }
     allocated_size_ = compute_aligned_size(size_t(shape_[layout_[0]]) * size_t(strides_[layout_[0]]), size_t(alignment));