diff options
Diffstat (limited to 'src/local_reductions.h')
-rw-r--r-- | src/local_reductions.h | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/src/local_reductions.h b/src/local_reductions.h index 3e3eba8..2f6804b 100644 --- a/src/local_reductions.h +++ b/src/local_reductions.h @@ -23,7 +23,7 @@ extern "C" { #endif -#define ITERATE_ON_ARRAY(i,cctk_type, in_data, out_type, out_num, mask_on,input_array_offset, indices, sum_indices, max_iter, iter, flag, actual_indices,input_array_strides, input_array_min_subscripts,product) \ +#define ITERATE_ON_ARRAY(i,cctk_type, in_data, out_type, out_num, mask_on,input_array_offset, indices, sum_indices, max_iter, iter, flag, actual_indices,input_array_strides, input_array_min_subscripts,input_array_dims,product) \ { \ const cctk_type * typed_vdata = (const cctk_type *)(in_data); \ out_type inval; \ @@ -44,7 +44,7 @@ extern "C" { product = 1; \ for (j=k-1;j>=0;j--) \ { \ - product *= actual_iters_per_dim[j]; \ + product *= input_array_dims[j]; \ } \ sum_indices += actual_indices[k]*product; \ } \ @@ -59,7 +59,7 @@ extern "C" { { \ if (flag==1) \ { \ - actual_indices[k] += input_array_strides[k-1]; \ + actual_indices[k] += input_array_strides[k]; \ indices[k]++; \ flag = 0; \ break; \ |