RubyGems - carray - Versions diffs - 3.0.5 → 3.0.6 - Mend

carray 3.0.5 → 3.0.6

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Files changed (119) hide show

checksums.yaml +4 -4
data/CHANGELOG.md +233 -0
data/README.md +1 -1
data/carray.gemspec +3 -2
data/ext/ca_axis_dispatch.c +6 -0
data/ext/ca_axis_group.c +1 -1
data/ext/ca_categorical_iterator.c +7 -7
data/ext/ca_kernel_iterator.c +116 -231
data/ext/ca_kernel_iterator.h +410 -254
data/ext/ca_obj_bincmp.c +9 -1
data/ext/ca_obj_binop.c +10 -1
data/ext/ca_obj_bitfield.c +2 -0
data/ext/ca_obj_byte_swap.c +2 -0
data/ext/ca_obj_face.c +2 -0
data/ext/ca_obj_fake.c +3 -1
data/ext/ca_obj_grid.c +167 -12
data/ext/ca_obj_meld.c +69 -113
data/ext/ca_obj_moncmp.c +9 -1
data/ext/ca_obj_monop.c +40 -2
data/ext/ca_obj_object.c +54 -20
data/ext/ca_obj_reduce.c +15 -1
data/ext/ca_obj_refer.c +65 -20
data/ext/ca_obj_remap.c +69 -10
data/ext/ca_obj_roll.c +1 -0
data/ext/ca_obj_select.c +4 -1
data/ext/ca_obj_select_axis.c +1 -0
data/ext/ca_obj_shift.c +1 -0
data/ext/ca_obj_stack.c +310 -111
data/ext/ca_obj_stride.c +131 -45
data/ext/ca_obj_string.c +18 -2
data/ext/ca_obj_tile.c +1 -0
data/ext/ca_obj_triop.c +9 -1
data/ext/ca_obj_window.c +1 -0
data/ext/ca_op_ipower.c +8 -8
data/ext/caframe_csv_reader.c +1 -1
data/ext/carray.h +2 -1
data/ext/carray_access.c +27 -64
data/ext/carray_address_basis.c +134 -0
data/ext/carray_attribute.c +8 -0
data/ext/carray_broadcast.c +3 -4
data/ext/carray_build_flags.h +1 -1
data/ext/carray_cast.c +8 -11
data/ext/carray_conversion.c +9 -5
data/ext/carray_core.c +811 -72
data/ext/carray_element.c +13 -9
data/ext/carray_histogram.c +25 -3
data/ext/carray_internal.h +147 -12
data/ext/carray_kernels_bincmp.c +2007 -3
data/ext/carray_kernels_binop.c +25365 -5199
data/ext/carray_kernels_init.c +8 -8
data/ext/carray_kernels_monop.c +10696 -3678
data/ext/carray_kernels_reduce_aggregate.c +4429 -702
data/ext/carray_kernels_reduce_boolean.c +1533 -15
data/ext/carray_kernels_reduce_cumulative.c +1808 -248
data/ext/carray_kernels_reduce_extreme.c +4508 -511
data/ext/carray_kernels_reduce_variance.c +2284 -152
data/ext/carray_kernels_scan.c +240 -240
data/ext/carray_kernels_search.c +4211 -264
data/ext/carray_kernels_sort.c +367 -450
data/ext/carray_lazy.c +47 -0
data/ext/carray_loop.c +1 -1
data/ext/carray_mask.c +130 -139
data/ext/carray_median_percentile.c +4 -4
data/ext/carray_memory_view.c +13 -7
data/ext/carray_operator.c +147 -44
data/ext/carray_order.c +3 -2
data/ext/carray_random.c +4 -1
data/ext/carray_slab.c +97 -21
data/ext/carray_sort.c +1 -1
data/ext/carray_test.c +74 -19
data/ext/carray_time_text.c +2 -2
data/ext/extconf.rb +11 -0
data/ext/mkkernel.rb +346 -94
data/ext/ruby_carray.c +4 -1
data/ext/version.h +4 -4
data/lib/carray/attribute.rb +25 -1
data/lib/carray/autoload_carray.rb +3 -3
data/lib/carray/autoload_method_extension.rb +5 -3
data/lib/carray/axis_group.rb +13 -7
data/lib/carray/basics.rb +1 -1
data/lib/carray/block_iterator.rb +17 -3
data/lib/carray/categorical.rb +6 -8
data/lib/carray/categorical_iterator.rb +2 -2
data/lib/carray/complex.rb +2 -2
data/lib/carray/conditional.rb +3 -2
data/lib/carray/frame/column_verbs.rb +45 -14
data/lib/carray/frame/convert.rb +19 -0
data/lib/carray/frame/csv_parser.rb +32 -32
data/lib/carray/frame/describe.rb +4 -2
data/lib/carray/frame/frame.rb +399 -78
data/lib/carray/frame/group.rb +679 -53
data/lib/carray/frame/io.rb +209 -71
data/lib/carray/frame/join.rb +146 -37
data/lib/carray/frame/records.rb +3 -3
data/lib/carray/frame/reshape.rb +89 -11
data/lib/carray/frame/sort.rb +57 -9
data/lib/carray/frame/verbs.rb +384 -49
data/lib/carray/frame/windows.rb +44 -6
data/lib/carray/inspect.rb +7 -2
data/lib/carray/iterator.rb +32 -0
data/lib/carray/lazy.rb +17 -8
data/lib/carray/mask_gap_fill.rb +37 -17
data/lib/carray/meld_reduce.rb +1 -1
data/lib/carray/methods/choose.rb +3 -2
data/lib/carray/methods/meshgrid.rb +3 -1
data/lib/carray/runtime.rb +47 -8
data/lib/carray/segment_iterator.rb +23 -8
data/lib/carray/serialize.rb +34 -17
data/lib/carray/stack.rb +18 -6
data/lib/carray/time.rb +5 -0
data/lib/carray/window_iterator.rb +16 -4
data/yard-stubs/ca_obj_array.rb +26 -1
data/yard-stubs/carray_cast.rb +8 -5
data/yard-stubs/carray_conversion.rb +8 -2
data/yard-stubs/carray_mask.rb +20 -1
data/yard-stubs/carray_math.rb +31 -16
data/yard-stubs/carray_memory_view.rb +7 -0
data/yard-stubs/carray_test.rb +7 -4
metadata +18 -4

data/ext/carray_kernels_sort.c CHANGED Viewed

@@ -63,6 +63,13 @@ sort_index_ki_native_i8 (VALUE self, CArray *ca, int axis, int do_stable, int ma
      positions align between iterators. */
   int8_t slab_axes[CA_RANK_MAX];
   slab_axes[0] = (int8_t) axis;
+  ca_size_t fiber_n = ca->dim[axis];
+  volatile VALUE hbuf = 0, haux = 0;
+  sort_index_pair_i8 *buf = ALLOCV_N(sort_index_pair_i8, hbuf, fiber_n);
+  sort_index_pair_i8 *aux = NULL;
+  if ( do_stable ) {
+    aux = ALLOCV_N(sort_index_pair_i8, haux, fiber_n);
+  }
   ca_iter_state st_in, st_out;
   int rc;
   rc = ca_iter_state_init_l2(&st_in,  ca, CA_SLAB_AXES,
@@ -77,14 +84,6 @@ sort_index_ki_native_i8 (VALUE self, CArray *ca, int axis, int do_stable, int ma
     rb_raise(rb_eRuntimeError, "%s: output init failed rc=%d", ca_calling_method_name(), rc);
   }
-  ca_size_t fiber_n = st_in.slab_elements;
-  volatile VALUE hbuf = 0, haux = 0;
-  sort_index_pair_i8 *buf = ALLOCV_N(sort_index_pair_i8, hbuf, fiber_n);
-  sort_index_pair_i8 *aux = NULL;
-  if ( do_stable ) {
-    aux = ALLOCV_N(sort_index_pair_i8, haux, fiber_n);
-  }
   char       *pi, *po;
   boolean8_t *mi, *mo;
   while ( ca_iter_state_next_slab_axes(&st_in,  &pi, &mi) &&
@@ -178,6 +177,13 @@ sort_index_ki_native_u8 (VALUE self, CArray *ca, int axis, int do_stable, int ma
      positions align between iterators. */
   int8_t slab_axes[CA_RANK_MAX];
   slab_axes[0] = (int8_t) axis;
+  ca_size_t fiber_n = ca->dim[axis];
+  volatile VALUE hbuf = 0, haux = 0;
+  sort_index_pair_u8 *buf = ALLOCV_N(sort_index_pair_u8, hbuf, fiber_n);
+  sort_index_pair_u8 *aux = NULL;
+  if ( do_stable ) {
+    aux = ALLOCV_N(sort_index_pair_u8, haux, fiber_n);
+  }
   ca_iter_state st_in, st_out;
   int rc;
   rc = ca_iter_state_init_l2(&st_in,  ca, CA_SLAB_AXES,
@@ -192,14 +198,6 @@ sort_index_ki_native_u8 (VALUE self, CArray *ca, int axis, int do_stable, int ma
     rb_raise(rb_eRuntimeError, "%s: output init failed rc=%d", ca_calling_method_name(), rc);
   }
-  ca_size_t fiber_n = st_in.slab_elements;
-  volatile VALUE hbuf = 0, haux = 0;
-  sort_index_pair_u8 *buf = ALLOCV_N(sort_index_pair_u8, hbuf, fiber_n);
-  sort_index_pair_u8 *aux = NULL;
-  if ( do_stable ) {
-    aux = ALLOCV_N(sort_index_pair_u8, haux, fiber_n);
-  }
   char       *pi, *po;
   boolean8_t *mi, *mo;
   while ( ca_iter_state_next_slab_axes(&st_in,  &pi, &mi) &&
@@ -293,6 +291,13 @@ sort_index_ki_native_i16 (VALUE self, CArray *ca, int axis, int do_stable, int m
      positions align between iterators. */
   int8_t slab_axes[CA_RANK_MAX];
   slab_axes[0] = (int8_t) axis;
+  ca_size_t fiber_n = ca->dim[axis];
+  volatile VALUE hbuf = 0, haux = 0;
+  sort_index_pair_i16 *buf = ALLOCV_N(sort_index_pair_i16, hbuf, fiber_n);
+  sort_index_pair_i16 *aux = NULL;
+  if ( do_stable ) {
+    aux = ALLOCV_N(sort_index_pair_i16, haux, fiber_n);
+  }
   ca_iter_state st_in, st_out;
   int rc;
   rc = ca_iter_state_init_l2(&st_in,  ca, CA_SLAB_AXES,
@@ -307,14 +312,6 @@ sort_index_ki_native_i16 (VALUE self, CArray *ca, int axis, int do_stable, int m
     rb_raise(rb_eRuntimeError, "%s: output init failed rc=%d", ca_calling_method_name(), rc);
   }
-  ca_size_t fiber_n = st_in.slab_elements;
-  volatile VALUE hbuf = 0, haux = 0;
-  sort_index_pair_i16 *buf = ALLOCV_N(sort_index_pair_i16, hbuf, fiber_n);
-  sort_index_pair_i16 *aux = NULL;
-  if ( do_stable ) {
-    aux = ALLOCV_N(sort_index_pair_i16, haux, fiber_n);
-  }
   char       *pi, *po;
   boolean8_t *mi, *mo;
   while ( ca_iter_state_next_slab_axes(&st_in,  &pi, &mi) &&
@@ -408,6 +405,13 @@ sort_index_ki_native_u16 (VALUE self, CArray *ca, int axis, int do_stable, int m
      positions align between iterators. */
   int8_t slab_axes[CA_RANK_MAX];
   slab_axes[0] = (int8_t) axis;
+  ca_size_t fiber_n = ca->dim[axis];
+  volatile VALUE hbuf = 0, haux = 0;
+  sort_index_pair_u16 *buf = ALLOCV_N(sort_index_pair_u16, hbuf, fiber_n);
+  sort_index_pair_u16 *aux = NULL;
+  if ( do_stable ) {
+    aux = ALLOCV_N(sort_index_pair_u16, haux, fiber_n);
+  }
   ca_iter_state st_in, st_out;
   int rc;
   rc = ca_iter_state_init_l2(&st_in,  ca, CA_SLAB_AXES,
@@ -422,14 +426,6 @@ sort_index_ki_native_u16 (VALUE self, CArray *ca, int axis, int do_stable, int m
     rb_raise(rb_eRuntimeError, "%s: output init failed rc=%d", ca_calling_method_name(), rc);
   }
-  ca_size_t fiber_n = st_in.slab_elements;
-  volatile VALUE hbuf = 0, haux = 0;
-  sort_index_pair_u16 *buf = ALLOCV_N(sort_index_pair_u16, hbuf, fiber_n);
-  sort_index_pair_u16 *aux = NULL;
-  if ( do_stable ) {
-    aux = ALLOCV_N(sort_index_pair_u16, haux, fiber_n);
-  }
   char       *pi, *po;
   boolean8_t *mi, *mo;
   while ( ca_iter_state_next_slab_axes(&st_in,  &pi, &mi) &&
@@ -523,6 +519,13 @@ sort_index_ki_native_i32 (VALUE self, CArray *ca, int axis, int do_stable, int m
      positions align between iterators. */
   int8_t slab_axes[CA_RANK_MAX];
   slab_axes[0] = (int8_t) axis;
+  ca_size_t fiber_n = ca->dim[axis];
+  volatile VALUE hbuf = 0, haux = 0;
+  sort_index_pair_i32 *buf = ALLOCV_N(sort_index_pair_i32, hbuf, fiber_n);
+  sort_index_pair_i32 *aux = NULL;
+  if ( do_stable ) {
+    aux = ALLOCV_N(sort_index_pair_i32, haux, fiber_n);
+  }
   ca_iter_state st_in, st_out;
   int rc;
   rc = ca_iter_state_init_l2(&st_in,  ca, CA_SLAB_AXES,
@@ -537,14 +540,6 @@ sort_index_ki_native_i32 (VALUE self, CArray *ca, int axis, int do_stable, int m
     rb_raise(rb_eRuntimeError, "%s: output init failed rc=%d", ca_calling_method_name(), rc);
   }
-  ca_size_t fiber_n = st_in.slab_elements;
-  volatile VALUE hbuf = 0, haux = 0;
-  sort_index_pair_i32 *buf = ALLOCV_N(sort_index_pair_i32, hbuf, fiber_n);
-  sort_index_pair_i32 *aux = NULL;
-  if ( do_stable ) {
-    aux = ALLOCV_N(sort_index_pair_i32, haux, fiber_n);
-  }
   char       *pi, *po;
   boolean8_t *mi, *mo;
   while ( ca_iter_state_next_slab_axes(&st_in,  &pi, &mi) &&
@@ -638,6 +633,13 @@ sort_index_ki_native_u32 (VALUE self, CArray *ca, int axis, int do_stable, int m
      positions align between iterators. */
   int8_t slab_axes[CA_RANK_MAX];
   slab_axes[0] = (int8_t) axis;
+  ca_size_t fiber_n = ca->dim[axis];
+  volatile VALUE hbuf = 0, haux = 0;
+  sort_index_pair_u32 *buf = ALLOCV_N(sort_index_pair_u32, hbuf, fiber_n);
+  sort_index_pair_u32 *aux = NULL;
+  if ( do_stable ) {
+    aux = ALLOCV_N(sort_index_pair_u32, haux, fiber_n);
+  }
   ca_iter_state st_in, st_out;
   int rc;
   rc = ca_iter_state_init_l2(&st_in,  ca, CA_SLAB_AXES,
@@ -652,14 +654,6 @@ sort_index_ki_native_u32 (VALUE self, CArray *ca, int axis, int do_stable, int m
     rb_raise(rb_eRuntimeError, "%s: output init failed rc=%d", ca_calling_method_name(), rc);
   }
-  ca_size_t fiber_n = st_in.slab_elements;
-  volatile VALUE hbuf = 0, haux = 0;
-  sort_index_pair_u32 *buf = ALLOCV_N(sort_index_pair_u32, hbuf, fiber_n);
-  sort_index_pair_u32 *aux = NULL;
-  if ( do_stable ) {
-    aux = ALLOCV_N(sort_index_pair_u32, haux, fiber_n);
-  }
   char       *pi, *po;
   boolean8_t *mi, *mo;
   while ( ca_iter_state_next_slab_axes(&st_in,  &pi, &mi) &&
@@ -753,6 +747,13 @@ sort_index_ki_native_i64 (VALUE self, CArray *ca, int axis, int do_stable, int m
      positions align between iterators. */
   int8_t slab_axes[CA_RANK_MAX];
   slab_axes[0] = (int8_t) axis;
+  ca_size_t fiber_n = ca->dim[axis];
+  volatile VALUE hbuf = 0, haux = 0;
+  sort_index_pair_i64 *buf = ALLOCV_N(sort_index_pair_i64, hbuf, fiber_n);
+  sort_index_pair_i64 *aux = NULL;
+  if ( do_stable ) {
+    aux = ALLOCV_N(sort_index_pair_i64, haux, fiber_n);
+  }
   ca_iter_state st_in, st_out;
   int rc;
   rc = ca_iter_state_init_l2(&st_in,  ca, CA_SLAB_AXES,
@@ -767,14 +768,6 @@ sort_index_ki_native_i64 (VALUE self, CArray *ca, int axis, int do_stable, int m
     rb_raise(rb_eRuntimeError, "%s: output init failed rc=%d", ca_calling_method_name(), rc);
   }
-  ca_size_t fiber_n = st_in.slab_elements;
-  volatile VALUE hbuf = 0, haux = 0;
-  sort_index_pair_i64 *buf = ALLOCV_N(sort_index_pair_i64, hbuf, fiber_n);
-  sort_index_pair_i64 *aux = NULL;
-  if ( do_stable ) {
-    aux = ALLOCV_N(sort_index_pair_i64, haux, fiber_n);
-  }
   char       *pi, *po;
   boolean8_t *mi, *mo;
   while ( ca_iter_state_next_slab_axes(&st_in,  &pi, &mi) &&
@@ -868,6 +861,13 @@ sort_index_ki_native_u64 (VALUE self, CArray *ca, int axis, int do_stable, int m
      positions align between iterators. */
   int8_t slab_axes[CA_RANK_MAX];
   slab_axes[0] = (int8_t) axis;
+  ca_size_t fiber_n = ca->dim[axis];
+  volatile VALUE hbuf = 0, haux = 0;
+  sort_index_pair_u64 *buf = ALLOCV_N(sort_index_pair_u64, hbuf, fiber_n);
+  sort_index_pair_u64 *aux = NULL;
+  if ( do_stable ) {
+    aux = ALLOCV_N(sort_index_pair_u64, haux, fiber_n);
+  }
   ca_iter_state st_in, st_out;
   int rc;
   rc = ca_iter_state_init_l2(&st_in,  ca, CA_SLAB_AXES,
@@ -882,14 +882,6 @@ sort_index_ki_native_u64 (VALUE self, CArray *ca, int axis, int do_stable, int m
     rb_raise(rb_eRuntimeError, "%s: output init failed rc=%d", ca_calling_method_name(), rc);
   }
-  ca_size_t fiber_n = st_in.slab_elements;
-  volatile VALUE hbuf = 0, haux = 0;
-  sort_index_pair_u64 *buf = ALLOCV_N(sort_index_pair_u64, hbuf, fiber_n);
-  sort_index_pair_u64 *aux = NULL;
-  if ( do_stable ) {
-    aux = ALLOCV_N(sort_index_pair_u64, haux, fiber_n);
-  }
   char       *pi, *po;
   boolean8_t *mi, *mo;
   while ( ca_iter_state_next_slab_axes(&st_in,  &pi, &mi) &&
@@ -983,6 +975,10 @@ sort_index_ki_native_f32 (VALUE self, CArray *ca, int axis, int do_stable, int m
      positions align between iterators. */
   int8_t slab_axes[CA_RANK_MAX];
   slab_axes[0] = (int8_t) axis;
+  ca_size_t fiber_n = ca->dim[axis];
+  volatile VALUE hbuf = 0, haux = 0;
+  sort_index_pair_f32 *buf = ALLOCV_N(sort_index_pair_f32, hbuf, fiber_n);
+  sort_index_pair_f32 *aux = ALLOCV_N(sort_index_pair_f32, haux, fiber_n);
   ca_iter_state st_in, st_out;
   int rc;
   rc = ca_iter_state_init_l2(&st_in,  ca, CA_SLAB_AXES,
@@ -997,14 +993,6 @@ sort_index_ki_native_f32 (VALUE self, CArray *ca, int axis, int do_stable, int m
     rb_raise(rb_eRuntimeError, "%s: output init failed rc=%d", ca_calling_method_name(), rc);
   }
-  ca_size_t fiber_n = st_in.slab_elements;
-  volatile VALUE hbuf = 0, haux = 0;
-  sort_index_pair_f32 *buf = ALLOCV_N(sort_index_pair_f32, hbuf, fiber_n);
-  sort_index_pair_f32 *aux = NULL;
-  if ( do_stable ) {
-    aux = ALLOCV_N(sort_index_pair_f32, haux, fiber_n);
-  }
   char       *pi, *po;
   boolean8_t *mi, *mo;
   while ( ca_iter_state_next_slab_axes(&st_in,  &pi, &mi) &&
@@ -1065,7 +1053,7 @@ sort_index_ki_native_f32 (VALUE self, CArray *ca, int axis, int do_stable, int m
     /* P.3/P.4: NaN pre-partition for float argsort -- NaN pairs are
        pushed to the tail (= NaN-at-end policy, matches sort_copy
        AC8).  Only the finite slice is sorted. */
-    ca_size_t finite_n = ca_partition_nan_pair_f32((ca_pair_f32 *) (buf + sort_lo), (ca_size_t) sort_n);
+    ca_size_t finite_n = ca_partition_nan_pair_f32((ca_pair_f32 *) (buf + sort_lo), (ca_size_t) sort_n, (ca_pair_f32 *) aux);
     if ( do_stable ) {
       ca_sort_merge_pair_f32((ca_pair_f32 *) (buf + sort_lo), (ca_pair_f32 *) aux, finite_n);
     } else {
@@ -1102,6 +1090,10 @@ sort_index_ki_native_f64 (VALUE self, CArray *ca, int axis, int do_stable, int m
      positions align between iterators. */
   int8_t slab_axes[CA_RANK_MAX];
   slab_axes[0] = (int8_t) axis;
+  ca_size_t fiber_n = ca->dim[axis];
+  volatile VALUE hbuf = 0, haux = 0;
+  sort_index_pair_f64 *buf = ALLOCV_N(sort_index_pair_f64, hbuf, fiber_n);
+  sort_index_pair_f64 *aux = ALLOCV_N(sort_index_pair_f64, haux, fiber_n);
   ca_iter_state st_in, st_out;
   int rc;
   rc = ca_iter_state_init_l2(&st_in,  ca, CA_SLAB_AXES,
@@ -1116,14 +1108,6 @@ sort_index_ki_native_f64 (VALUE self, CArray *ca, int axis, int do_stable, int m
     rb_raise(rb_eRuntimeError, "%s: output init failed rc=%d", ca_calling_method_name(), rc);
   }
-  ca_size_t fiber_n = st_in.slab_elements;
-  volatile VALUE hbuf = 0, haux = 0;
-  sort_index_pair_f64 *buf = ALLOCV_N(sort_index_pair_f64, hbuf, fiber_n);
-  sort_index_pair_f64 *aux = NULL;
-  if ( do_stable ) {
-    aux = ALLOCV_N(sort_index_pair_f64, haux, fiber_n);
-  }
   char       *pi, *po;
   boolean8_t *mi, *mo;
   while ( ca_iter_state_next_slab_axes(&st_in,  &pi, &mi) &&
@@ -1184,7 +1168,7 @@ sort_index_ki_native_f64 (VALUE self, CArray *ca, int axis, int do_stable, int m
     /* P.3/P.4: NaN pre-partition for float argsort -- NaN pairs are
        pushed to the tail (= NaN-at-end policy, matches sort_copy
        AC8).  Only the finite slice is sorted. */
-    ca_size_t finite_n = ca_partition_nan_pair_f64((ca_pair_f64 *) (buf + sort_lo), (ca_size_t) sort_n);
+    ca_size_t finite_n = ca_partition_nan_pair_f64((ca_pair_f64 *) (buf + sort_lo), (ca_size_t) sort_n, (ca_pair_f64 *) aux);
     if ( do_stable ) {
       ca_sort_merge_pair_f64((ca_pair_f64 *) (buf + sort_lo), (ca_pair_f64 *) aux, finite_n);
     } else {
@@ -1221,6 +1205,13 @@ sort_index_ki_native_bool (VALUE self, CArray *ca, int axis, int do_stable, int
      positions align between iterators. */
   int8_t slab_axes[CA_RANK_MAX];
   slab_axes[0] = (int8_t) axis;
+  ca_size_t fiber_n = ca->dim[axis];
+  volatile VALUE hbuf = 0, haux = 0;
+  sort_index_pair_bool *buf = ALLOCV_N(sort_index_pair_bool, hbuf, fiber_n);
+  sort_index_pair_bool *aux = NULL;
+  if ( do_stable ) {
+    aux = ALLOCV_N(sort_index_pair_bool, haux, fiber_n);
+  }
   ca_iter_state st_in, st_out;
   int rc;
   rc = ca_iter_state_init_l2(&st_in,  ca, CA_SLAB_AXES,
@@ -1235,14 +1226,6 @@ sort_index_ki_native_bool (VALUE self, CArray *ca, int axis, int do_stable, int
     rb_raise(rb_eRuntimeError, "%s: output init failed rc=%d", ca_calling_method_name(), rc);
   }
-  ca_size_t fiber_n = st_in.slab_elements;
-  volatile VALUE hbuf = 0, haux = 0;
-  sort_index_pair_bool *buf = ALLOCV_N(sort_index_pair_bool, hbuf, fiber_n);
-  sort_index_pair_bool *aux = NULL;
-  if ( do_stable ) {
-    aux = ALLOCV_N(sort_index_pair_bool, haux, fiber_n);
-  }
   char       *pi, *po;
   boolean8_t *mi, *mo;
   while ( ca_iter_state_next_slab_axes(&st_in,  &pi, &mi) &&
@@ -1336,9 +1319,9 @@ sort_index_cmp_object (const void *a, const void *b)
   return 0;
 }
-/* The object lane calls Ruby for every cell, so a walk can raise part
-   way.  Its states live in the caller's frame, and ca_iter_ensure
-   finishes them however the body leaves. */
+/* The walks can raise part way (a Ruby call per cell in the object
+   lane, an allocation in any lane).  Their states live in the caller's
+   frame, and ca_iter_ensure finishes them however the body leaves. */
 typedef struct {
   ca_iter_state states[2];
   VALUE self;
@@ -1374,6 +1357,10 @@ sort_index_ki_native_object_body (VALUE arg)
      positions align between iterators. */
   int8_t slab_axes[CA_RANK_MAX];
   slab_axes[0] = (int8_t) axis;
+  ca_size_t fiber_n = ca->dim[axis];
+  volatile VALUE hbuf = 0, haux = 0;
+  sort_index_pair_object *buf = ALLOCV_N(sort_index_pair_object, hbuf, fiber_n);
+  sort_index_pair_object *aux = ALLOCV_N(sort_index_pair_object, haux, fiber_n);
   int rc;
   rc = ca_iter_state_init_l2(&st_in,  ca, CA_SLAB_AXES,
                              slab_axes, 1, 0);
@@ -1387,11 +1374,6 @@ sort_index_ki_native_object_body (VALUE arg)
     rb_raise(rb_eRuntimeError, "%s: output init failed rc=%d", ca_calling_method_name(), rc);
   }
-  ca_size_t fiber_n = st_in.slab_elements;
-  volatile VALUE hbuf = 0, haux = 0;
-  sort_index_pair_object *buf = ALLOCV_N(sort_index_pair_object, hbuf, fiber_n);
-  sort_index_pair_object *aux = ALLOCV_N(sort_index_pair_object, haux, fiber_n);
   char       *pi, *po;
   boolean8_t *mi, *mo;
   while ( ca_iter_state_next_slab_axes(&st_in,  &pi, &mi) &&
@@ -1475,7 +1457,7 @@ sort_index_ki_native_object (VALUE self, CArray *ca, int axis, int do_stable, in
   fr.axis = axis;
   fr.do_stable = do_stable;
   fr.masked_last = masked_last;
-  return ca_iter_ensure(2, fr.states, sort_index_ki_native_object_body, (VALUE) &fr);
+  return ca_iter_ensure_computing(ca, 2, fr.states, sort_index_ki_native_object_body, (VALUE) &fr);
 }
@@ -1511,6 +1493,10 @@ sort_index_ki_native_fixlen (VALUE self, CArray *ca, int axis, int do_stable, in
      positions align between iterators. */
   int8_t slab_axes[CA_RANK_MAX];
   slab_axes[0] = (int8_t) axis;
+  ca_size_t fiber_n = ca->dim[axis];
+  volatile VALUE hbuf = 0, haux = 0;
+  sort_index_pair_fixlen *buf = ALLOCV_N(sort_index_pair_fixlen, hbuf, fiber_n);
+  (void) haux;
   ca_iter_state st_in, st_out;
   int rc;
   rc = ca_iter_state_init_l2(&st_in,  ca, CA_SLAB_AXES,
@@ -1525,11 +1511,6 @@ sort_index_ki_native_fixlen (VALUE self, CArray *ca, int axis, int do_stable, in
     rb_raise(rb_eRuntimeError, "%s: output init failed rc=%d", ca_calling_method_name(), rc);
   }
-  ca_size_t fiber_n = st_in.slab_elements;
-  volatile VALUE hbuf = 0, haux = 0;
-  sort_index_pair_fixlen *buf = ALLOCV_N(sort_index_pair_fixlen, hbuf, fiber_n);
-  (void) haux;
   char       *pi, *po;
   boolean8_t *mi, *mo;
   while ( ca_iter_state_next_slab_axes(&st_in,  &pi, &mi) &&
@@ -1710,6 +1691,13 @@ sort_addr_ki_native_i8 (VALUE self, CArray *ca, int axis, int do_stable, int mas
      positions align between iterators. */
   int8_t slab_axes[CA_RANK_MAX];
   slab_axes[0] = (int8_t) axis;
+  ca_size_t fiber_n = ca->dim[axis];
+  volatile VALUE hbuf = 0, haux = 0;
+  sort_addr_pair_i8 *buf = ALLOCV_N(sort_addr_pair_i8, hbuf, fiber_n);
+  sort_addr_pair_i8 *aux = NULL;
+  if ( do_stable ) {
+    aux = ALLOCV_N(sort_addr_pair_i8, haux, fiber_n);
+  }
   ca_iter_state st_in, st_out;
   int rc;
   rc = ca_iter_state_init_l2(&st_in,  ca, CA_SLAB_AXES,
@@ -1724,14 +1712,6 @@ sort_addr_ki_native_i8 (VALUE self, CArray *ca, int axis, int do_stable, int mas
     rb_raise(rb_eRuntimeError, "%s: output init failed rc=%d", ca_calling_method_name(), rc);
   }
-  ca_size_t fiber_n = st_in.slab_elements;
-  volatile VALUE hbuf = 0, haux = 0;
-  sort_addr_pair_i8 *buf = ALLOCV_N(sort_addr_pair_i8, hbuf, fiber_n);
-  sort_addr_pair_i8 *aux = NULL;
-  if ( do_stable ) {
-    aux = ALLOCV_N(sort_addr_pair_i8, haux, fiber_n);
-  }
   /* :view_flat semantics: kernel computes view-flat address
      (= row-major position in ca's shape, 0..elements-1) for
      each cell.  Used to feed CARemap.idx directly (= sort
@@ -1857,6 +1837,13 @@ sort_addr_ki_native_u8 (VALUE self, CArray *ca, int axis, int do_stable, int mas
      positions align between iterators. */
   int8_t slab_axes[CA_RANK_MAX];
   slab_axes[0] = (int8_t) axis;
+  ca_size_t fiber_n = ca->dim[axis];
+  volatile VALUE hbuf = 0, haux = 0;
+  sort_addr_pair_u8 *buf = ALLOCV_N(sort_addr_pair_u8, hbuf, fiber_n);
+  sort_addr_pair_u8 *aux = NULL;
+  if ( do_stable ) {
+    aux = ALLOCV_N(sort_addr_pair_u8, haux, fiber_n);
+  }
   ca_iter_state st_in, st_out;
   int rc;
   rc = ca_iter_state_init_l2(&st_in,  ca, CA_SLAB_AXES,
@@ -1871,14 +1858,6 @@ sort_addr_ki_native_u8 (VALUE self, CArray *ca, int axis, int do_stable, int mas
     rb_raise(rb_eRuntimeError, "%s: output init failed rc=%d", ca_calling_method_name(), rc);
   }
-  ca_size_t fiber_n = st_in.slab_elements;
-  volatile VALUE hbuf = 0, haux = 0;
-  sort_addr_pair_u8 *buf = ALLOCV_N(sort_addr_pair_u8, hbuf, fiber_n);
-  sort_addr_pair_u8 *aux = NULL;
-  if ( do_stable ) {
-    aux = ALLOCV_N(sort_addr_pair_u8, haux, fiber_n);
-  }
   /* :view_flat semantics: kernel computes view-flat address
      (= row-major position in ca's shape, 0..elements-1) for
      each cell.  Used to feed CARemap.idx directly (= sort
@@ -2004,6 +1983,13 @@ sort_addr_ki_native_i16 (VALUE self, CArray *ca, int axis, int do_stable, int ma
      positions align between iterators. */
   int8_t slab_axes[CA_RANK_MAX];
   slab_axes[0] = (int8_t) axis;
+  ca_size_t fiber_n = ca->dim[axis];
+  volatile VALUE hbuf = 0, haux = 0;
+  sort_addr_pair_i16 *buf = ALLOCV_N(sort_addr_pair_i16, hbuf, fiber_n);
+  sort_addr_pair_i16 *aux = NULL;
+  if ( do_stable ) {
+    aux = ALLOCV_N(sort_addr_pair_i16, haux, fiber_n);
+  }
   ca_iter_state st_in, st_out;
   int rc;
   rc = ca_iter_state_init_l2(&st_in,  ca, CA_SLAB_AXES,
@@ -2018,14 +2004,6 @@ sort_addr_ki_native_i16 (VALUE self, CArray *ca, int axis, int do_stable, int ma
     rb_raise(rb_eRuntimeError, "%s: output init failed rc=%d", ca_calling_method_name(), rc);
   }
-  ca_size_t fiber_n = st_in.slab_elements;
-  volatile VALUE hbuf = 0, haux = 0;
-  sort_addr_pair_i16 *buf = ALLOCV_N(sort_addr_pair_i16, hbuf, fiber_n);
-  sort_addr_pair_i16 *aux = NULL;
-  if ( do_stable ) {
-    aux = ALLOCV_N(sort_addr_pair_i16, haux, fiber_n);
-  }
   /* :view_flat semantics: kernel computes view-flat address
      (= row-major position in ca's shape, 0..elements-1) for
      each cell.  Used to feed CARemap.idx directly (= sort
@@ -2151,6 +2129,13 @@ sort_addr_ki_native_u16 (VALUE self, CArray *ca, int axis, int do_stable, int ma
      positions align between iterators. */
   int8_t slab_axes[CA_RANK_MAX];
   slab_axes[0] = (int8_t) axis;
+  ca_size_t fiber_n = ca->dim[axis];
+  volatile VALUE hbuf = 0, haux = 0;
+  sort_addr_pair_u16 *buf = ALLOCV_N(sort_addr_pair_u16, hbuf, fiber_n);
+  sort_addr_pair_u16 *aux = NULL;
+  if ( do_stable ) {
+    aux = ALLOCV_N(sort_addr_pair_u16, haux, fiber_n);
+  }
   ca_iter_state st_in, st_out;
   int rc;
   rc = ca_iter_state_init_l2(&st_in,  ca, CA_SLAB_AXES,
@@ -2165,14 +2150,6 @@ sort_addr_ki_native_u16 (VALUE self, CArray *ca, int axis, int do_stable, int ma
     rb_raise(rb_eRuntimeError, "%s: output init failed rc=%d", ca_calling_method_name(), rc);
   }
-  ca_size_t fiber_n = st_in.slab_elements;
-  volatile VALUE hbuf = 0, haux = 0;
-  sort_addr_pair_u16 *buf = ALLOCV_N(sort_addr_pair_u16, hbuf, fiber_n);
-  sort_addr_pair_u16 *aux = NULL;
-  if ( do_stable ) {
-    aux = ALLOCV_N(sort_addr_pair_u16, haux, fiber_n);
-  }
   /* :view_flat semantics: kernel computes view-flat address
      (= row-major position in ca's shape, 0..elements-1) for
      each cell.  Used to feed CARemap.idx directly (= sort
@@ -2298,6 +2275,13 @@ sort_addr_ki_native_i32 (VALUE self, CArray *ca, int axis, int do_stable, int ma
      positions align between iterators. */
   int8_t slab_axes[CA_RANK_MAX];
   slab_axes[0] = (int8_t) axis;
+  ca_size_t fiber_n = ca->dim[axis];
+  volatile VALUE hbuf = 0, haux = 0;
+  sort_addr_pair_i32 *buf = ALLOCV_N(sort_addr_pair_i32, hbuf, fiber_n);
+  sort_addr_pair_i32 *aux = NULL;
+  if ( do_stable ) {
+    aux = ALLOCV_N(sort_addr_pair_i32, haux, fiber_n);
+  }
   ca_iter_state st_in, st_out;
   int rc;
   rc = ca_iter_state_init_l2(&st_in,  ca, CA_SLAB_AXES,
@@ -2312,14 +2296,6 @@ sort_addr_ki_native_i32 (VALUE self, CArray *ca, int axis, int do_stable, int ma
     rb_raise(rb_eRuntimeError, "%s: output init failed rc=%d", ca_calling_method_name(), rc);
   }
-  ca_size_t fiber_n = st_in.slab_elements;
-  volatile VALUE hbuf = 0, haux = 0;
-  sort_addr_pair_i32 *buf = ALLOCV_N(sort_addr_pair_i32, hbuf, fiber_n);
-  sort_addr_pair_i32 *aux = NULL;
-  if ( do_stable ) {
-    aux = ALLOCV_N(sort_addr_pair_i32, haux, fiber_n);
-  }
   /* :view_flat semantics: kernel computes view-flat address
      (= row-major position in ca's shape, 0..elements-1) for
      each cell.  Used to feed CARemap.idx directly (= sort
@@ -2445,6 +2421,13 @@ sort_addr_ki_native_u32 (VALUE self, CArray *ca, int axis, int do_stable, int ma
      positions align between iterators. */
   int8_t slab_axes[CA_RANK_MAX];
   slab_axes[0] = (int8_t) axis;
+  ca_size_t fiber_n = ca->dim[axis];
+  volatile VALUE hbuf = 0, haux = 0;
+  sort_addr_pair_u32 *buf = ALLOCV_N(sort_addr_pair_u32, hbuf, fiber_n);
+  sort_addr_pair_u32 *aux = NULL;
+  if ( do_stable ) {
+    aux = ALLOCV_N(sort_addr_pair_u32, haux, fiber_n);
+  }
   ca_iter_state st_in, st_out;
   int rc;
   rc = ca_iter_state_init_l2(&st_in,  ca, CA_SLAB_AXES,
@@ -2459,14 +2442,6 @@ sort_addr_ki_native_u32 (VALUE self, CArray *ca, int axis, int do_stable, int ma
     rb_raise(rb_eRuntimeError, "%s: output init failed rc=%d", ca_calling_method_name(), rc);
   }
-  ca_size_t fiber_n = st_in.slab_elements;
-  volatile VALUE hbuf = 0, haux = 0;
-  sort_addr_pair_u32 *buf = ALLOCV_N(sort_addr_pair_u32, hbuf, fiber_n);
-  sort_addr_pair_u32 *aux = NULL;
-  if ( do_stable ) {
-    aux = ALLOCV_N(sort_addr_pair_u32, haux, fiber_n);
-  }
   /* :view_flat semantics: kernel computes view-flat address
      (= row-major position in ca's shape, 0..elements-1) for
      each cell.  Used to feed CARemap.idx directly (= sort
@@ -2592,6 +2567,13 @@ sort_addr_ki_native_i64 (VALUE self, CArray *ca, int axis, int do_stable, int ma
      positions align between iterators. */
   int8_t slab_axes[CA_RANK_MAX];
   slab_axes[0] = (int8_t) axis;
+  ca_size_t fiber_n = ca->dim[axis];
+  volatile VALUE hbuf = 0, haux = 0;
+  sort_addr_pair_i64 *buf = ALLOCV_N(sort_addr_pair_i64, hbuf, fiber_n);
+  sort_addr_pair_i64 *aux = NULL;
+  if ( do_stable ) {
+    aux = ALLOCV_N(sort_addr_pair_i64, haux, fiber_n);
+  }
   ca_iter_state st_in, st_out;
   int rc;
   rc = ca_iter_state_init_l2(&st_in,  ca, CA_SLAB_AXES,
@@ -2606,14 +2588,6 @@ sort_addr_ki_native_i64 (VALUE self, CArray *ca, int axis, int do_stable, int ma
     rb_raise(rb_eRuntimeError, "%s: output init failed rc=%d", ca_calling_method_name(), rc);
   }
-  ca_size_t fiber_n = st_in.slab_elements;
-  volatile VALUE hbuf = 0, haux = 0;
-  sort_addr_pair_i64 *buf = ALLOCV_N(sort_addr_pair_i64, hbuf, fiber_n);
-  sort_addr_pair_i64 *aux = NULL;
-  if ( do_stable ) {
-    aux = ALLOCV_N(sort_addr_pair_i64, haux, fiber_n);
-  }
   /* :view_flat semantics: kernel computes view-flat address
      (= row-major position in ca's shape, 0..elements-1) for
      each cell.  Used to feed CARemap.idx directly (= sort
@@ -2739,6 +2713,13 @@ sort_addr_ki_native_u64 (VALUE self, CArray *ca, int axis, int do_stable, int ma
      positions align between iterators. */
   int8_t slab_axes[CA_RANK_MAX];
   slab_axes[0] = (int8_t) axis;
+  ca_size_t fiber_n = ca->dim[axis];
+  volatile VALUE hbuf = 0, haux = 0;
+  sort_addr_pair_u64 *buf = ALLOCV_N(sort_addr_pair_u64, hbuf, fiber_n);
+  sort_addr_pair_u64 *aux = NULL;
+  if ( do_stable ) {
+    aux = ALLOCV_N(sort_addr_pair_u64, haux, fiber_n);
+  }
   ca_iter_state st_in, st_out;
   int rc;
   rc = ca_iter_state_init_l2(&st_in,  ca, CA_SLAB_AXES,
@@ -2753,14 +2734,6 @@ sort_addr_ki_native_u64 (VALUE self, CArray *ca, int axis, int do_stable, int ma
     rb_raise(rb_eRuntimeError, "%s: output init failed rc=%d", ca_calling_method_name(), rc);
   }
-  ca_size_t fiber_n = st_in.slab_elements;
-  volatile VALUE hbuf = 0, haux = 0;
-  sort_addr_pair_u64 *buf = ALLOCV_N(sort_addr_pair_u64, hbuf, fiber_n);
-  sort_addr_pair_u64 *aux = NULL;
-  if ( do_stable ) {
-    aux = ALLOCV_N(sort_addr_pair_u64, haux, fiber_n);
-  }
   /* :view_flat semantics: kernel computes view-flat address
      (= row-major position in ca's shape, 0..elements-1) for
      each cell.  Used to feed CARemap.idx directly (= sort
@@ -2886,6 +2859,10 @@ sort_addr_ki_native_f32 (VALUE self, CArray *ca, int axis, int do_stable, int ma
      positions align between iterators. */
   int8_t slab_axes[CA_RANK_MAX];
   slab_axes[0] = (int8_t) axis;
+  ca_size_t fiber_n = ca->dim[axis];
+  volatile VALUE hbuf = 0, haux = 0;
+  sort_addr_pair_f32 *buf = ALLOCV_N(sort_addr_pair_f32, hbuf, fiber_n);
+  sort_addr_pair_f32 *aux = ALLOCV_N(sort_addr_pair_f32, haux, fiber_n);
   ca_iter_state st_in, st_out;
   int rc;
   rc = ca_iter_state_init_l2(&st_in,  ca, CA_SLAB_AXES,
@@ -2900,14 +2877,6 @@ sort_addr_ki_native_f32 (VALUE self, CArray *ca, int axis, int do_stable, int ma
     rb_raise(rb_eRuntimeError, "%s: output init failed rc=%d", ca_calling_method_name(), rc);
   }
-  ca_size_t fiber_n = st_in.slab_elements;
-  volatile VALUE hbuf = 0, haux = 0;
-  sort_addr_pair_f32 *buf = ALLOCV_N(sort_addr_pair_f32, hbuf, fiber_n);
-  sort_addr_pair_f32 *aux = NULL;
-  if ( do_stable ) {
-    aux = ALLOCV_N(sort_addr_pair_f32, haux, fiber_n);
-  }
   /* :view_flat semantics: kernel computes view-flat address
      (= row-major position in ca's shape, 0..elements-1) for
      each cell.  Used to feed CARemap.idx directly (= sort
@@ -2995,7 +2964,7 @@ sort_addr_ki_native_f32 (VALUE self, CArray *ca, int axis, int do_stable, int ma
     /* P.3/P.4: NaN pre-partition for float argsort -- NaN pairs are
        pushed to the tail (= NaN-at-end policy, matches sort_copy
        AC8).  Only the finite slice is sorted. */
-    ca_size_t finite_n = ca_partition_nan_pair_f32((ca_pair_f32 *) (buf + sort_lo), (ca_size_t) sort_n);
+    ca_size_t finite_n = ca_partition_nan_pair_f32((ca_pair_f32 *) (buf + sort_lo), (ca_size_t) sort_n, (ca_pair_f32 *) aux);
     if ( do_stable ) {
       ca_sort_merge_pair_f32((ca_pair_f32 *) (buf + sort_lo), (ca_pair_f32 *) aux, finite_n);
     } else {
@@ -3037,6 +3006,10 @@ sort_addr_ki_native_f64 (VALUE self, CArray *ca, int axis, int do_stable, int ma
      positions align between iterators. */
   int8_t slab_axes[CA_RANK_MAX];
   slab_axes[0] = (int8_t) axis;
+  ca_size_t fiber_n = ca->dim[axis];
+  volatile VALUE hbuf = 0, haux = 0;
+  sort_addr_pair_f64 *buf = ALLOCV_N(sort_addr_pair_f64, hbuf, fiber_n);
+  sort_addr_pair_f64 *aux = ALLOCV_N(sort_addr_pair_f64, haux, fiber_n);
   ca_iter_state st_in, st_out;
   int rc;
   rc = ca_iter_state_init_l2(&st_in,  ca, CA_SLAB_AXES,
@@ -3051,14 +3024,6 @@ sort_addr_ki_native_f64 (VALUE self, CArray *ca, int axis, int do_stable, int ma
     rb_raise(rb_eRuntimeError, "%s: output init failed rc=%d", ca_calling_method_name(), rc);
   }
-  ca_size_t fiber_n = st_in.slab_elements;
-  volatile VALUE hbuf = 0, haux = 0;
-  sort_addr_pair_f64 *buf = ALLOCV_N(sort_addr_pair_f64, hbuf, fiber_n);
-  sort_addr_pair_f64 *aux = NULL;
-  if ( do_stable ) {
-    aux = ALLOCV_N(sort_addr_pair_f64, haux, fiber_n);
-  }
   /* :view_flat semantics: kernel computes view-flat address
      (= row-major position in ca's shape, 0..elements-1) for
      each cell.  Used to feed CARemap.idx directly (= sort
@@ -3146,7 +3111,7 @@ sort_addr_ki_native_f64 (VALUE self, CArray *ca, int axis, int do_stable, int ma
     /* P.3/P.4: NaN pre-partition for float argsort -- NaN pairs are
        pushed to the tail (= NaN-at-end policy, matches sort_copy
        AC8).  Only the finite slice is sorted. */
-    ca_size_t finite_n = ca_partition_nan_pair_f64((ca_pair_f64 *) (buf + sort_lo), (ca_size_t) sort_n);
+    ca_size_t finite_n = ca_partition_nan_pair_f64((ca_pair_f64 *) (buf + sort_lo), (ca_size_t) sort_n, (ca_pair_f64 *) aux);
     if ( do_stable ) {
       ca_sort_merge_pair_f64((ca_pair_f64 *) (buf + sort_lo), (ca_pair_f64 *) aux, finite_n);
     } else {
@@ -3188,6 +3153,13 @@ sort_addr_ki_native_bool (VALUE self, CArray *ca, int axis, int do_stable, int m
      positions align between iterators. */
   int8_t slab_axes[CA_RANK_MAX];
   slab_axes[0] = (int8_t) axis;
+  ca_size_t fiber_n = ca->dim[axis];
+  volatile VALUE hbuf = 0, haux = 0;
+  sort_addr_pair_bool *buf = ALLOCV_N(sort_addr_pair_bool, hbuf, fiber_n);
+  sort_addr_pair_bool *aux = NULL;
+  if ( do_stable ) {
+    aux = ALLOCV_N(sort_addr_pair_bool, haux, fiber_n);
+  }
   ca_iter_state st_in, st_out;
   int rc;
   rc = ca_iter_state_init_l2(&st_in,  ca, CA_SLAB_AXES,
@@ -3202,14 +3174,6 @@ sort_addr_ki_native_bool (VALUE self, CArray *ca, int axis, int do_stable, int m
     rb_raise(rb_eRuntimeError, "%s: output init failed rc=%d", ca_calling_method_name(), rc);
   }
-  ca_size_t fiber_n = st_in.slab_elements;
-  volatile VALUE hbuf = 0, haux = 0;
-  sort_addr_pair_bool *buf = ALLOCV_N(sort_addr_pair_bool, hbuf, fiber_n);
-  sort_addr_pair_bool *aux = NULL;
-  if ( do_stable ) {
-    aux = ALLOCV_N(sort_addr_pair_bool, haux, fiber_n);
-  }
   /* :view_flat semantics: kernel computes view-flat address
      (= row-major position in ca's shape, 0..elements-1) for
      each cell.  Used to feed CARemap.idx directly (= sort
@@ -3335,9 +3299,9 @@ sort_addr_cmp_object (const void *a, const void *b)
   return 0;
 }
-/* The object lane calls Ruby for every cell, so a walk can raise part
-   way.  Its states live in the caller's frame, and ca_iter_ensure
-   finishes them however the body leaves. */
+/* The walks can raise part way (a Ruby call per cell in the object
+   lane, an allocation in any lane).  Their states live in the caller's
+   frame, and ca_iter_ensure finishes them however the body leaves. */
 typedef struct {
   ca_iter_state states[2];
   VALUE self;
@@ -3373,6 +3337,10 @@ sort_addr_ki_native_object_body (VALUE arg)
      positions align between iterators. */
   int8_t slab_axes[CA_RANK_MAX];
   slab_axes[0] = (int8_t) axis;
+  ca_size_t fiber_n = ca->dim[axis];
+  volatile VALUE hbuf = 0, haux = 0;
+  sort_addr_pair_object *buf = ALLOCV_N(sort_addr_pair_object, hbuf, fiber_n);
+  sort_addr_pair_object *aux = ALLOCV_N(sort_addr_pair_object, haux, fiber_n);
   int rc;
   rc = ca_iter_state_init_l2(&st_in,  ca, CA_SLAB_AXES,
                              slab_axes, 1, 0);
@@ -3386,11 +3354,6 @@ sort_addr_ki_native_object_body (VALUE arg)
     rb_raise(rb_eRuntimeError, "%s: output init failed rc=%d", ca_calling_method_name(), rc);
   }
-  ca_size_t fiber_n = st_in.slab_elements;
-  volatile VALUE hbuf = 0, haux = 0;
-  sort_addr_pair_object *buf = ALLOCV_N(sort_addr_pair_object, hbuf, fiber_n);
-  sort_addr_pair_object *aux = ALLOCV_N(sort_addr_pair_object, haux, fiber_n);
   /* :view_flat semantics: kernel computes view-flat address
      (= row-major position in ca's shape, 0..elements-1) for
      each cell.  Used to feed CARemap.idx directly (= sort
@@ -3506,7 +3469,7 @@ sort_addr_ki_native_object (VALUE self, CArray *ca, int axis, int do_stable, int
   fr.axis = axis;
   fr.do_stable = do_stable;
   fr.masked_last = masked_last;
-  return ca_iter_ensure(2, fr.states, sort_addr_ki_native_object_body, (VALUE) &fr);
+  return ca_iter_ensure_computing(ca, 2, fr.states, sort_addr_ki_native_object_body, (VALUE) &fr);
 }
@@ -3542,6 +3505,10 @@ sort_addr_ki_native_fixlen (VALUE self, CArray *ca, int axis, int do_stable, int
      positions align between iterators. */
   int8_t slab_axes[CA_RANK_MAX];
   slab_axes[0] = (int8_t) axis;
+  ca_size_t fiber_n = ca->dim[axis];
+  volatile VALUE hbuf = 0, haux = 0;
+  sort_addr_pair_fixlen *buf = ALLOCV_N(sort_addr_pair_fixlen, hbuf, fiber_n);
+  (void) haux;
   ca_iter_state st_in, st_out;
   int rc;
   rc = ca_iter_state_init_l2(&st_in,  ca, CA_SLAB_AXES,
@@ -3556,11 +3523,6 @@ sort_addr_ki_native_fixlen (VALUE self, CArray *ca, int axis, int do_stable, int
     rb_raise(rb_eRuntimeError, "%s: output init failed rc=%d", ca_calling_method_name(), rc);
   }
-  ca_size_t fiber_n = st_in.slab_elements;
-  volatile VALUE hbuf = 0, haux = 0;
-  sort_addr_pair_fixlen *buf = ALLOCV_N(sort_addr_pair_fixlen, hbuf, fiber_n);
-  (void) haux;
   /* :view_flat semantics: kernel computes view-flat address
      (= row-major position in ca's shape, 0..elements-1) for
      each cell.  Used to feed CARemap.idx directly (= sort
@@ -3811,6 +3773,10 @@ partition_index_ki_native_i8 (VALUE self, CArray *ca, int axis, ca_size_t kth, i
      positions align between iterators. */
   int8_t slab_axes[CA_RANK_MAX];
   slab_axes[0] = (int8_t) axis;
+  ca_size_t fiber_n = ca->dim[axis];
+  volatile VALUE hbuf = 0, haux = 0;
+  partition_index_pair_i8 *buf = ALLOCV_N(partition_index_pair_i8, hbuf, fiber_n);
+  (void) haux;
   ca_iter_state st_in, st_out;
   int rc;
   rc = ca_iter_state_init_l2(&st_in,  ca, CA_SLAB_AXES,
@@ -3825,11 +3791,6 @@ partition_index_ki_native_i8 (VALUE self, CArray *ca, int axis, ca_size_t kth, i
     rb_raise(rb_eRuntimeError, "%s: output init failed rc=%d", ca_calling_method_name(), rc);
   }
-  ca_size_t fiber_n = st_in.slab_elements;
-  volatile VALUE hbuf = 0, haux = 0;
-  partition_index_pair_i8 *buf = ALLOCV_N(partition_index_pair_i8, hbuf, fiber_n);
-  (void) haux;
   char       *pi, *po;
   boolean8_t *mi, *mo;
   while ( ca_iter_state_next_slab_axes(&st_in,  &pi, &mi) &&
@@ -3990,6 +3951,10 @@ partition_index_ki_native_u8 (VALUE self, CArray *ca, int axis, ca_size_t kth, i
      positions align between iterators. */
   int8_t slab_axes[CA_RANK_MAX];
   slab_axes[0] = (int8_t) axis;
+  ca_size_t fiber_n = ca->dim[axis];
+  volatile VALUE hbuf = 0, haux = 0;
+  partition_index_pair_u8 *buf = ALLOCV_N(partition_index_pair_u8, hbuf, fiber_n);
+  (void) haux;
   ca_iter_state st_in, st_out;
   int rc;
   rc = ca_iter_state_init_l2(&st_in,  ca, CA_SLAB_AXES,
@@ -4004,11 +3969,6 @@ partition_index_ki_native_u8 (VALUE self, CArray *ca, int axis, ca_size_t kth, i
     rb_raise(rb_eRuntimeError, "%s: output init failed rc=%d", ca_calling_method_name(), rc);
   }
-  ca_size_t fiber_n = st_in.slab_elements;
-  volatile VALUE hbuf = 0, haux = 0;
-  partition_index_pair_u8 *buf = ALLOCV_N(partition_index_pair_u8, hbuf, fiber_n);
-  (void) haux;
   char       *pi, *po;
   boolean8_t *mi, *mo;
   while ( ca_iter_state_next_slab_axes(&st_in,  &pi, &mi) &&
@@ -4169,6 +4129,10 @@ partition_index_ki_native_i16 (VALUE self, CArray *ca, int axis, ca_size_t kth,
      positions align between iterators. */
   int8_t slab_axes[CA_RANK_MAX];
   slab_axes[0] = (int8_t) axis;
+  ca_size_t fiber_n = ca->dim[axis];
+  volatile VALUE hbuf = 0, haux = 0;
+  partition_index_pair_i16 *buf = ALLOCV_N(partition_index_pair_i16, hbuf, fiber_n);
+  (void) haux;
   ca_iter_state st_in, st_out;
   int rc;
   rc = ca_iter_state_init_l2(&st_in,  ca, CA_SLAB_AXES,
@@ -4183,11 +4147,6 @@ partition_index_ki_native_i16 (VALUE self, CArray *ca, int axis, ca_size_t kth,
     rb_raise(rb_eRuntimeError, "%s: output init failed rc=%d", ca_calling_method_name(), rc);
   }
-  ca_size_t fiber_n = st_in.slab_elements;
-  volatile VALUE hbuf = 0, haux = 0;
-  partition_index_pair_i16 *buf = ALLOCV_N(partition_index_pair_i16, hbuf, fiber_n);
-  (void) haux;
   char       *pi, *po;
   boolean8_t *mi, *mo;
   while ( ca_iter_state_next_slab_axes(&st_in,  &pi, &mi) &&
@@ -4348,6 +4307,10 @@ partition_index_ki_native_u16 (VALUE self, CArray *ca, int axis, ca_size_t kth,
      positions align between iterators. */
   int8_t slab_axes[CA_RANK_MAX];
   slab_axes[0] = (int8_t) axis;
+  ca_size_t fiber_n = ca->dim[axis];
+  volatile VALUE hbuf = 0, haux = 0;
+  partition_index_pair_u16 *buf = ALLOCV_N(partition_index_pair_u16, hbuf, fiber_n);
+  (void) haux;
   ca_iter_state st_in, st_out;
   int rc;
   rc = ca_iter_state_init_l2(&st_in,  ca, CA_SLAB_AXES,
@@ -4362,11 +4325,6 @@ partition_index_ki_native_u16 (VALUE self, CArray *ca, int axis, ca_size_t kth,
     rb_raise(rb_eRuntimeError, "%s: output init failed rc=%d", ca_calling_method_name(), rc);
   }
-  ca_size_t fiber_n = st_in.slab_elements;
-  volatile VALUE hbuf = 0, haux = 0;
-  partition_index_pair_u16 *buf = ALLOCV_N(partition_index_pair_u16, hbuf, fiber_n);
-  (void) haux;
   char       *pi, *po;
   boolean8_t *mi, *mo;
   while ( ca_iter_state_next_slab_axes(&st_in,  &pi, &mi) &&
@@ -4527,6 +4485,10 @@ partition_index_ki_native_i32 (VALUE self, CArray *ca, int axis, ca_size_t kth,
      positions align between iterators. */
   int8_t slab_axes[CA_RANK_MAX];
   slab_axes[0] = (int8_t) axis;
+  ca_size_t fiber_n = ca->dim[axis];
+  volatile VALUE hbuf = 0, haux = 0;
+  partition_index_pair_i32 *buf = ALLOCV_N(partition_index_pair_i32, hbuf, fiber_n);
+  (void) haux;
   ca_iter_state st_in, st_out;
   int rc;
   rc = ca_iter_state_init_l2(&st_in,  ca, CA_SLAB_AXES,
@@ -4541,11 +4503,6 @@ partition_index_ki_native_i32 (VALUE self, CArray *ca, int axis, ca_size_t kth,
     rb_raise(rb_eRuntimeError, "%s: output init failed rc=%d", ca_calling_method_name(), rc);
   }
-  ca_size_t fiber_n = st_in.slab_elements;
-  volatile VALUE hbuf = 0, haux = 0;
-  partition_index_pair_i32 *buf = ALLOCV_N(partition_index_pair_i32, hbuf, fiber_n);
-  (void) haux;
   char       *pi, *po;
   boolean8_t *mi, *mo;
   while ( ca_iter_state_next_slab_axes(&st_in,  &pi, &mi) &&
@@ -4706,6 +4663,10 @@ partition_index_ki_native_u32 (VALUE self, CArray *ca, int axis, ca_size_t kth,
      positions align between iterators. */
   int8_t slab_axes[CA_RANK_MAX];
   slab_axes[0] = (int8_t) axis;
+  ca_size_t fiber_n = ca->dim[axis];
+  volatile VALUE hbuf = 0, haux = 0;
+  partition_index_pair_u32 *buf = ALLOCV_N(partition_index_pair_u32, hbuf, fiber_n);
+  (void) haux;
   ca_iter_state st_in, st_out;
   int rc;
   rc = ca_iter_state_init_l2(&st_in,  ca, CA_SLAB_AXES,
@@ -4720,11 +4681,6 @@ partition_index_ki_native_u32 (VALUE self, CArray *ca, int axis, ca_size_t kth,
     rb_raise(rb_eRuntimeError, "%s: output init failed rc=%d", ca_calling_method_name(), rc);
   }
-  ca_size_t fiber_n = st_in.slab_elements;
-  volatile VALUE hbuf = 0, haux = 0;
-  partition_index_pair_u32 *buf = ALLOCV_N(partition_index_pair_u32, hbuf, fiber_n);
-  (void) haux;
   char       *pi, *po;
   boolean8_t *mi, *mo;
   while ( ca_iter_state_next_slab_axes(&st_in,  &pi, &mi) &&
@@ -4885,6 +4841,10 @@ partition_index_ki_native_i64 (VALUE self, CArray *ca, int axis, ca_size_t kth,
      positions align between iterators. */
   int8_t slab_axes[CA_RANK_MAX];
   slab_axes[0] = (int8_t) axis;
+  ca_size_t fiber_n = ca->dim[axis];
+  volatile VALUE hbuf = 0, haux = 0;
+  partition_index_pair_i64 *buf = ALLOCV_N(partition_index_pair_i64, hbuf, fiber_n);
+  (void) haux;
   ca_iter_state st_in, st_out;
   int rc;
   rc = ca_iter_state_init_l2(&st_in,  ca, CA_SLAB_AXES,
@@ -4899,11 +4859,6 @@ partition_index_ki_native_i64 (VALUE self, CArray *ca, int axis, ca_size_t kth,
     rb_raise(rb_eRuntimeError, "%s: output init failed rc=%d", ca_calling_method_name(), rc);
   }
-  ca_size_t fiber_n = st_in.slab_elements;
-  volatile VALUE hbuf = 0, haux = 0;
-  partition_index_pair_i64 *buf = ALLOCV_N(partition_index_pair_i64, hbuf, fiber_n);
-  (void) haux;
   char       *pi, *po;
   boolean8_t *mi, *mo;
   while ( ca_iter_state_next_slab_axes(&st_in,  &pi, &mi) &&
@@ -5064,6 +5019,10 @@ partition_index_ki_native_u64 (VALUE self, CArray *ca, int axis, ca_size_t kth,
      positions align between iterators. */
   int8_t slab_axes[CA_RANK_MAX];
   slab_axes[0] = (int8_t) axis;
+  ca_size_t fiber_n = ca->dim[axis];
+  volatile VALUE hbuf = 0, haux = 0;
+  partition_index_pair_u64 *buf = ALLOCV_N(partition_index_pair_u64, hbuf, fiber_n);
+  (void) haux;
   ca_iter_state st_in, st_out;
   int rc;
   rc = ca_iter_state_init_l2(&st_in,  ca, CA_SLAB_AXES,
@@ -5078,11 +5037,6 @@ partition_index_ki_native_u64 (VALUE self, CArray *ca, int axis, ca_size_t kth,
     rb_raise(rb_eRuntimeError, "%s: output init failed rc=%d", ca_calling_method_name(), rc);
   }
-  ca_size_t fiber_n = st_in.slab_elements;
-  volatile VALUE hbuf = 0, haux = 0;
-  partition_index_pair_u64 *buf = ALLOCV_N(partition_index_pair_u64, hbuf, fiber_n);
-  (void) haux;
   char       *pi, *po;
   boolean8_t *mi, *mo;
   while ( ca_iter_state_next_slab_axes(&st_in,  &pi, &mi) &&
@@ -5243,6 +5197,10 @@ partition_index_ki_native_f32 (VALUE self, CArray *ca, int axis, ca_size_t kth,
      positions align between iterators. */
   int8_t slab_axes[CA_RANK_MAX];
   slab_axes[0] = (int8_t) axis;
+  ca_size_t fiber_n = ca->dim[axis];
+  volatile VALUE hbuf = 0, haux = 0;
+  partition_index_pair_f32 *buf = ALLOCV_N(partition_index_pair_f32, hbuf, fiber_n);
+  (void) haux;
   ca_iter_state st_in, st_out;
   int rc;
   rc = ca_iter_state_init_l2(&st_in,  ca, CA_SLAB_AXES,
@@ -5257,11 +5215,6 @@ partition_index_ki_native_f32 (VALUE self, CArray *ca, int axis, ca_size_t kth,
     rb_raise(rb_eRuntimeError, "%s: output init failed rc=%d", ca_calling_method_name(), rc);
   }
-  ca_size_t fiber_n = st_in.slab_elements;
-  volatile VALUE hbuf = 0, haux = 0;
-  partition_index_pair_f32 *buf = ALLOCV_N(partition_index_pair_f32, hbuf, fiber_n);
-  (void) haux;
   char       *pi, *po;
   boolean8_t *mi, *mo;
   while ( ca_iter_state_next_slab_axes(&st_in,  &pi, &mi) &&
@@ -5422,6 +5375,10 @@ partition_index_ki_native_f64 (VALUE self, CArray *ca, int axis, ca_size_t kth,
      positions align between iterators. */
   int8_t slab_axes[CA_RANK_MAX];
   slab_axes[0] = (int8_t) axis;
+  ca_size_t fiber_n = ca->dim[axis];
+  volatile VALUE hbuf = 0, haux = 0;
+  partition_index_pair_f64 *buf = ALLOCV_N(partition_index_pair_f64, hbuf, fiber_n);
+  (void) haux;
   ca_iter_state st_in, st_out;
   int rc;
   rc = ca_iter_state_init_l2(&st_in,  ca, CA_SLAB_AXES,
@@ -5436,11 +5393,6 @@ partition_index_ki_native_f64 (VALUE self, CArray *ca, int axis, ca_size_t kth,
     rb_raise(rb_eRuntimeError, "%s: output init failed rc=%d", ca_calling_method_name(), rc);
   }
-  ca_size_t fiber_n = st_in.slab_elements;
-  volatile VALUE hbuf = 0, haux = 0;
-  partition_index_pair_f64 *buf = ALLOCV_N(partition_index_pair_f64, hbuf, fiber_n);
-  (void) haux;
   char       *pi, *po;
   boolean8_t *mi, *mo;
   while ( ca_iter_state_next_slab_axes(&st_in,  &pi, &mi) &&
@@ -5601,6 +5553,10 @@ partition_index_ki_native_bool (VALUE self, CArray *ca, int axis, ca_size_t kth,
      positions align between iterators. */
   int8_t slab_axes[CA_RANK_MAX];
   slab_axes[0] = (int8_t) axis;
+  ca_size_t fiber_n = ca->dim[axis];
+  volatile VALUE hbuf = 0, haux = 0;
+  partition_index_pair_bool *buf = ALLOCV_N(partition_index_pair_bool, hbuf, fiber_n);
+  (void) haux;
   ca_iter_state st_in, st_out;
   int rc;
   rc = ca_iter_state_init_l2(&st_in,  ca, CA_SLAB_AXES,
@@ -5615,11 +5571,6 @@ partition_index_ki_native_bool (VALUE self, CArray *ca, int axis, ca_size_t kth,
     rb_raise(rb_eRuntimeError, "%s: output init failed rc=%d", ca_calling_method_name(), rc);
   }
-  ca_size_t fiber_n = st_in.slab_elements;
-  volatile VALUE hbuf = 0, haux = 0;
-  partition_index_pair_bool *buf = ALLOCV_N(partition_index_pair_bool, hbuf, fiber_n);
-  (void) haux;
   char       *pi, *po;
   boolean8_t *mi, *mo;
   while ( ca_iter_state_next_slab_axes(&st_in,  &pi, &mi) &&
@@ -5767,9 +5718,9 @@ partition_index_quickselect_object (partition_index_pair_object *arr,
   }
 }
-/* The object lane calls Ruby for every cell, so a walk can raise part
-   way.  Its states live in the caller's frame, and ca_iter_ensure
-   finishes them however the body leaves. */
+/* The walks can raise part way (a Ruby call per cell in the object
+   lane, an allocation in any lane).  Their states live in the caller's
+   frame, and ca_iter_ensure finishes them however the body leaves. */
 typedef struct {
   ca_iter_state states[2];
   VALUE self;
@@ -5805,6 +5756,10 @@ partition_index_ki_native_object_body (VALUE arg)
      positions align between iterators. */
   int8_t slab_axes[CA_RANK_MAX];
   slab_axes[0] = (int8_t) axis;
+  ca_size_t fiber_n = ca->dim[axis];
+  volatile VALUE hbuf = 0, haux = 0;
+  partition_index_pair_object *buf = ALLOCV_N(partition_index_pair_object, hbuf, fiber_n);
+  (void) haux;
   int rc;
   rc = ca_iter_state_init_l2(&st_in,  ca, CA_SLAB_AXES,
                              slab_axes, 1, 0);
@@ -5818,11 +5773,6 @@ partition_index_ki_native_object_body (VALUE arg)
     rb_raise(rb_eRuntimeError, "%s: output init failed rc=%d", ca_calling_method_name(), rc);
   }
-  ca_size_t fiber_n = st_in.slab_elements;
-  volatile VALUE hbuf = 0, haux = 0;
-  partition_index_pair_object *buf = ALLOCV_N(partition_index_pair_object, hbuf, fiber_n);
-  (void) haux;
   char       *pi, *po;
   boolean8_t *mi, *mo;
   while ( ca_iter_state_next_slab_axes(&st_in,  &pi, &mi) &&
@@ -5908,7 +5858,7 @@ partition_index_ki_native_object (VALUE self, CArray *ca, int axis, ca_size_t kt
   fr.axis = axis;
   fr.kth = kth;
   fr.masked_last = masked_last;
-  return ca_iter_ensure(2, fr.states, partition_index_ki_native_object_body, (VALUE) &fr);
+  return ca_iter_ensure_computing(ca, 2, fr.states, partition_index_ki_native_object_body, (VALUE) &fr);
 }
@@ -5999,6 +5949,10 @@ partition_index_ki_native_fixlen (VALUE self, CArray *ca, int axis, ca_size_t kt
      positions align between iterators. */
   int8_t slab_axes[CA_RANK_MAX];
   slab_axes[0] = (int8_t) axis;
+  ca_size_t fiber_n = ca->dim[axis];
+  volatile VALUE hbuf = 0, haux = 0;
+  partition_index_pair_fixlen *buf = ALLOCV_N(partition_index_pair_fixlen, hbuf, fiber_n);
+  (void) haux;
   ca_iter_state st_in, st_out;
   int rc;
   rc = ca_iter_state_init_l2(&st_in,  ca, CA_SLAB_AXES,
@@ -6013,11 +5967,6 @@ partition_index_ki_native_fixlen (VALUE self, CArray *ca, int axis, ca_size_t kt
     rb_raise(rb_eRuntimeError, "%s: output init failed rc=%d", ca_calling_method_name(), rc);
   }
-  ca_size_t fiber_n = st_in.slab_elements;
-  volatile VALUE hbuf = 0, haux = 0;
-  partition_index_pair_fixlen *buf = ALLOCV_N(partition_index_pair_fixlen, hbuf, fiber_n);
-  (void) haux;
   char       *pi, *po;
   boolean8_t *mi, *mo;
   while ( ca_iter_state_next_slab_axes(&st_in,  &pi, &mi) &&
@@ -6254,6 +6203,10 @@ partition_addr_ki_native_i8 (VALUE self, CArray *ca, int axis, ca_size_t kth, in
      positions align between iterators. */
   int8_t slab_axes[CA_RANK_MAX];
   slab_axes[0] = (int8_t) axis;
+  ca_size_t fiber_n = ca->dim[axis];
+  volatile VALUE hbuf = 0, haux = 0;
+  partition_addr_pair_i8 *buf = ALLOCV_N(partition_addr_pair_i8, hbuf, fiber_n);
+  (void) haux;
   ca_iter_state st_in, st_out;
   int rc;
   rc = ca_iter_state_init_l2(&st_in,  ca, CA_SLAB_AXES,
@@ -6268,11 +6221,6 @@ partition_addr_ki_native_i8 (VALUE self, CArray *ca, int axis, ca_size_t kth, in
     rb_raise(rb_eRuntimeError, "%s: output init failed rc=%d", ca_calling_method_name(), rc);
   }
-  ca_size_t fiber_n = st_in.slab_elements;
-  volatile VALUE hbuf = 0, haux = 0;
-  partition_addr_pair_i8 *buf = ALLOCV_N(partition_addr_pair_i8, hbuf, fiber_n);
-  (void) haux;
   /* :view_flat semantics: kernel computes view-flat address
      (= row-major position in ca's shape, 0..elements-1) for
      each cell.  Used to feed CARemap.idx directly (= sort
@@ -6465,6 +6413,10 @@ partition_addr_ki_native_u8 (VALUE self, CArray *ca, int axis, ca_size_t kth, in
      positions align between iterators. */
   int8_t slab_axes[CA_RANK_MAX];
   slab_axes[0] = (int8_t) axis;
+  ca_size_t fiber_n = ca->dim[axis];
+  volatile VALUE hbuf = 0, haux = 0;
+  partition_addr_pair_u8 *buf = ALLOCV_N(partition_addr_pair_u8, hbuf, fiber_n);
+  (void) haux;
   ca_iter_state st_in, st_out;
   int rc;
   rc = ca_iter_state_init_l2(&st_in,  ca, CA_SLAB_AXES,
@@ -6479,11 +6431,6 @@ partition_addr_ki_native_u8 (VALUE self, CArray *ca, int axis, ca_size_t kth, in
     rb_raise(rb_eRuntimeError, "%s: output init failed rc=%d", ca_calling_method_name(), rc);
   }
-  ca_size_t fiber_n = st_in.slab_elements;
-  volatile VALUE hbuf = 0, haux = 0;
-  partition_addr_pair_u8 *buf = ALLOCV_N(partition_addr_pair_u8, hbuf, fiber_n);
-  (void) haux;
   /* :view_flat semantics: kernel computes view-flat address
      (= row-major position in ca's shape, 0..elements-1) for
      each cell.  Used to feed CARemap.idx directly (= sort
@@ -6676,6 +6623,10 @@ partition_addr_ki_native_i16 (VALUE self, CArray *ca, int axis, ca_size_t kth, i
      positions align between iterators. */
   int8_t slab_axes[CA_RANK_MAX];
   slab_axes[0] = (int8_t) axis;
+  ca_size_t fiber_n = ca->dim[axis];
+  volatile VALUE hbuf = 0, haux = 0;
+  partition_addr_pair_i16 *buf = ALLOCV_N(partition_addr_pair_i16, hbuf, fiber_n);
+  (void) haux;
   ca_iter_state st_in, st_out;
   int rc;
   rc = ca_iter_state_init_l2(&st_in,  ca, CA_SLAB_AXES,
@@ -6690,11 +6641,6 @@ partition_addr_ki_native_i16 (VALUE self, CArray *ca, int axis, ca_size_t kth, i
     rb_raise(rb_eRuntimeError, "%s: output init failed rc=%d", ca_calling_method_name(), rc);
   }
-  ca_size_t fiber_n = st_in.slab_elements;
-  volatile VALUE hbuf = 0, haux = 0;
-  partition_addr_pair_i16 *buf = ALLOCV_N(partition_addr_pair_i16, hbuf, fiber_n);
-  (void) haux;
   /* :view_flat semantics: kernel computes view-flat address
      (= row-major position in ca's shape, 0..elements-1) for
      each cell.  Used to feed CARemap.idx directly (= sort
@@ -6887,6 +6833,10 @@ partition_addr_ki_native_u16 (VALUE self, CArray *ca, int axis, ca_size_t kth, i
      positions align between iterators. */
   int8_t slab_axes[CA_RANK_MAX];
   slab_axes[0] = (int8_t) axis;
+  ca_size_t fiber_n = ca->dim[axis];
+  volatile VALUE hbuf = 0, haux = 0;
+  partition_addr_pair_u16 *buf = ALLOCV_N(partition_addr_pair_u16, hbuf, fiber_n);
+  (void) haux;
   ca_iter_state st_in, st_out;
   int rc;
   rc = ca_iter_state_init_l2(&st_in,  ca, CA_SLAB_AXES,
@@ -6901,11 +6851,6 @@ partition_addr_ki_native_u16 (VALUE self, CArray *ca, int axis, ca_size_t kth, i
     rb_raise(rb_eRuntimeError, "%s: output init failed rc=%d", ca_calling_method_name(), rc);
   }
-  ca_size_t fiber_n = st_in.slab_elements;
-  volatile VALUE hbuf = 0, haux = 0;
-  partition_addr_pair_u16 *buf = ALLOCV_N(partition_addr_pair_u16, hbuf, fiber_n);
-  (void) haux;
   /* :view_flat semantics: kernel computes view-flat address
      (= row-major position in ca's shape, 0..elements-1) for
      each cell.  Used to feed CARemap.idx directly (= sort
@@ -7098,6 +7043,10 @@ partition_addr_ki_native_i32 (VALUE self, CArray *ca, int axis, ca_size_t kth, i
      positions align between iterators. */
   int8_t slab_axes[CA_RANK_MAX];
   slab_axes[0] = (int8_t) axis;
+  ca_size_t fiber_n = ca->dim[axis];
+  volatile VALUE hbuf = 0, haux = 0;
+  partition_addr_pair_i32 *buf = ALLOCV_N(partition_addr_pair_i32, hbuf, fiber_n);
+  (void) haux;
   ca_iter_state st_in, st_out;
   int rc;
   rc = ca_iter_state_init_l2(&st_in,  ca, CA_SLAB_AXES,
@@ -7112,11 +7061,6 @@ partition_addr_ki_native_i32 (VALUE self, CArray *ca, int axis, ca_size_t kth, i
     rb_raise(rb_eRuntimeError, "%s: output init failed rc=%d", ca_calling_method_name(), rc);
   }
-  ca_size_t fiber_n = st_in.slab_elements;
-  volatile VALUE hbuf = 0, haux = 0;
-  partition_addr_pair_i32 *buf = ALLOCV_N(partition_addr_pair_i32, hbuf, fiber_n);
-  (void) haux;
   /* :view_flat semantics: kernel computes view-flat address
      (= row-major position in ca's shape, 0..elements-1) for
      each cell.  Used to feed CARemap.idx directly (= sort
@@ -7309,6 +7253,10 @@ partition_addr_ki_native_u32 (VALUE self, CArray *ca, int axis, ca_size_t kth, i
      positions align between iterators. */
   int8_t slab_axes[CA_RANK_MAX];
   slab_axes[0] = (int8_t) axis;
+  ca_size_t fiber_n = ca->dim[axis];
+  volatile VALUE hbuf = 0, haux = 0;
+  partition_addr_pair_u32 *buf = ALLOCV_N(partition_addr_pair_u32, hbuf, fiber_n);
+  (void) haux;
   ca_iter_state st_in, st_out;
   int rc;
   rc = ca_iter_state_init_l2(&st_in,  ca, CA_SLAB_AXES,
@@ -7323,11 +7271,6 @@ partition_addr_ki_native_u32 (VALUE self, CArray *ca, int axis, ca_size_t kth, i
     rb_raise(rb_eRuntimeError, "%s: output init failed rc=%d", ca_calling_method_name(), rc);
   }
-  ca_size_t fiber_n = st_in.slab_elements;
-  volatile VALUE hbuf = 0, haux = 0;
-  partition_addr_pair_u32 *buf = ALLOCV_N(partition_addr_pair_u32, hbuf, fiber_n);
-  (void) haux;
   /* :view_flat semantics: kernel computes view-flat address
      (= row-major position in ca's shape, 0..elements-1) for
      each cell.  Used to feed CARemap.idx directly (= sort
@@ -7520,6 +7463,10 @@ partition_addr_ki_native_i64 (VALUE self, CArray *ca, int axis, ca_size_t kth, i
      positions align between iterators. */
   int8_t slab_axes[CA_RANK_MAX];
   slab_axes[0] = (int8_t) axis;
+  ca_size_t fiber_n = ca->dim[axis];
+  volatile VALUE hbuf = 0, haux = 0;
+  partition_addr_pair_i64 *buf = ALLOCV_N(partition_addr_pair_i64, hbuf, fiber_n);
+  (void) haux;
   ca_iter_state st_in, st_out;
   int rc;
   rc = ca_iter_state_init_l2(&st_in,  ca, CA_SLAB_AXES,
@@ -7534,11 +7481,6 @@ partition_addr_ki_native_i64 (VALUE self, CArray *ca, int axis, ca_size_t kth, i
     rb_raise(rb_eRuntimeError, "%s: output init failed rc=%d", ca_calling_method_name(), rc);
   }
-  ca_size_t fiber_n = st_in.slab_elements;
-  volatile VALUE hbuf = 0, haux = 0;
-  partition_addr_pair_i64 *buf = ALLOCV_N(partition_addr_pair_i64, hbuf, fiber_n);
-  (void) haux;
   /* :view_flat semantics: kernel computes view-flat address
      (= row-major position in ca's shape, 0..elements-1) for
      each cell.  Used to feed CARemap.idx directly (= sort
@@ -7731,6 +7673,10 @@ partition_addr_ki_native_u64 (VALUE self, CArray *ca, int axis, ca_size_t kth, i
      positions align between iterators. */
   int8_t slab_axes[CA_RANK_MAX];
   slab_axes[0] = (int8_t) axis;
+  ca_size_t fiber_n = ca->dim[axis];
+  volatile VALUE hbuf = 0, haux = 0;
+  partition_addr_pair_u64 *buf = ALLOCV_N(partition_addr_pair_u64, hbuf, fiber_n);
+  (void) haux;
   ca_iter_state st_in, st_out;
   int rc;
   rc = ca_iter_state_init_l2(&st_in,  ca, CA_SLAB_AXES,
@@ -7745,11 +7691,6 @@ partition_addr_ki_native_u64 (VALUE self, CArray *ca, int axis, ca_size_t kth, i
     rb_raise(rb_eRuntimeError, "%s: output init failed rc=%d", ca_calling_method_name(), rc);
   }
-  ca_size_t fiber_n = st_in.slab_elements;
-  volatile VALUE hbuf = 0, haux = 0;
-  partition_addr_pair_u64 *buf = ALLOCV_N(partition_addr_pair_u64, hbuf, fiber_n);
-  (void) haux;
   /* :view_flat semantics: kernel computes view-flat address
      (= row-major position in ca's shape, 0..elements-1) for
      each cell.  Used to feed CARemap.idx directly (= sort
@@ -7942,6 +7883,10 @@ partition_addr_ki_native_f32 (VALUE self, CArray *ca, int axis, ca_size_t kth, i
      positions align between iterators. */
   int8_t slab_axes[CA_RANK_MAX];
   slab_axes[0] = (int8_t) axis;
+  ca_size_t fiber_n = ca->dim[axis];
+  volatile VALUE hbuf = 0, haux = 0;
+  partition_addr_pair_f32 *buf = ALLOCV_N(partition_addr_pair_f32, hbuf, fiber_n);
+  (void) haux;
   ca_iter_state st_in, st_out;
   int rc;
   rc = ca_iter_state_init_l2(&st_in,  ca, CA_SLAB_AXES,
@@ -7956,11 +7901,6 @@ partition_addr_ki_native_f32 (VALUE self, CArray *ca, int axis, ca_size_t kth, i
     rb_raise(rb_eRuntimeError, "%s: output init failed rc=%d", ca_calling_method_name(), rc);
   }
-  ca_size_t fiber_n = st_in.slab_elements;
-  volatile VALUE hbuf = 0, haux = 0;
-  partition_addr_pair_f32 *buf = ALLOCV_N(partition_addr_pair_f32, hbuf, fiber_n);
-  (void) haux;
   /* :view_flat semantics: kernel computes view-flat address
      (= row-major position in ca's shape, 0..elements-1) for
      each cell.  Used to feed CARemap.idx directly (= sort
@@ -8153,6 +8093,10 @@ partition_addr_ki_native_f64 (VALUE self, CArray *ca, int axis, ca_size_t kth, i
      positions align between iterators. */
   int8_t slab_axes[CA_RANK_MAX];
   slab_axes[0] = (int8_t) axis;
+  ca_size_t fiber_n = ca->dim[axis];
+  volatile VALUE hbuf = 0, haux = 0;
+  partition_addr_pair_f64 *buf = ALLOCV_N(partition_addr_pair_f64, hbuf, fiber_n);
+  (void) haux;
   ca_iter_state st_in, st_out;
   int rc;
   rc = ca_iter_state_init_l2(&st_in,  ca, CA_SLAB_AXES,
@@ -8167,11 +8111,6 @@ partition_addr_ki_native_f64 (VALUE self, CArray *ca, int axis, ca_size_t kth, i
     rb_raise(rb_eRuntimeError, "%s: output init failed rc=%d", ca_calling_method_name(), rc);
   }
-  ca_size_t fiber_n = st_in.slab_elements;
-  volatile VALUE hbuf = 0, haux = 0;
-  partition_addr_pair_f64 *buf = ALLOCV_N(partition_addr_pair_f64, hbuf, fiber_n);
-  (void) haux;
   /* :view_flat semantics: kernel computes view-flat address
      (= row-major position in ca's shape, 0..elements-1) for
      each cell.  Used to feed CARemap.idx directly (= sort
@@ -8364,6 +8303,10 @@ partition_addr_ki_native_bool (VALUE self, CArray *ca, int axis, ca_size_t kth,
      positions align between iterators. */
   int8_t slab_axes[CA_RANK_MAX];
   slab_axes[0] = (int8_t) axis;
+  ca_size_t fiber_n = ca->dim[axis];
+  volatile VALUE hbuf = 0, haux = 0;
+  partition_addr_pair_bool *buf = ALLOCV_N(partition_addr_pair_bool, hbuf, fiber_n);
+  (void) haux;
   ca_iter_state st_in, st_out;
   int rc;
   rc = ca_iter_state_init_l2(&st_in,  ca, CA_SLAB_AXES,
@@ -8378,11 +8321,6 @@ partition_addr_ki_native_bool (VALUE self, CArray *ca, int axis, ca_size_t kth,
     rb_raise(rb_eRuntimeError, "%s: output init failed rc=%d", ca_calling_method_name(), rc);
   }
-  ca_size_t fiber_n = st_in.slab_elements;
-  volatile VALUE hbuf = 0, haux = 0;
-  partition_addr_pair_bool *buf = ALLOCV_N(partition_addr_pair_bool, hbuf, fiber_n);
-  (void) haux;
   /* :view_flat semantics: kernel computes view-flat address
      (= row-major position in ca's shape, 0..elements-1) for
      each cell.  Used to feed CARemap.idx directly (= sort
@@ -8562,9 +8500,9 @@ partition_addr_quickselect_object (partition_addr_pair_object *arr,
   }
 }
-/* The object lane calls Ruby for every cell, so a walk can raise part
-   way.  Its states live in the caller's frame, and ca_iter_ensure
-   finishes them however the body leaves. */
+/* The walks can raise part way (a Ruby call per cell in the object
+   lane, an allocation in any lane).  Their states live in the caller's
+   frame, and ca_iter_ensure finishes them however the body leaves. */
 typedef struct {
   ca_iter_state states[2];
   VALUE self;
@@ -8600,6 +8538,10 @@ partition_addr_ki_native_object_body (VALUE arg)
      positions align between iterators. */
   int8_t slab_axes[CA_RANK_MAX];
   slab_axes[0] = (int8_t) axis;
+  ca_size_t fiber_n = ca->dim[axis];
+  volatile VALUE hbuf = 0, haux = 0;
+  partition_addr_pair_object *buf = ALLOCV_N(partition_addr_pair_object, hbuf, fiber_n);
+  (void) haux;
   int rc;
   rc = ca_iter_state_init_l2(&st_in,  ca, CA_SLAB_AXES,
                              slab_axes, 1, 0);
@@ -8613,11 +8555,6 @@ partition_addr_ki_native_object_body (VALUE arg)
     rb_raise(rb_eRuntimeError, "%s: output init failed rc=%d", ca_calling_method_name(), rc);
   }
-  ca_size_t fiber_n = st_in.slab_elements;
-  volatile VALUE hbuf = 0, haux = 0;
-  partition_addr_pair_object *buf = ALLOCV_N(partition_addr_pair_object, hbuf, fiber_n);
-  (void) haux;
   /* :view_flat semantics: kernel computes view-flat address
      (= row-major position in ca's shape, 0..elements-1) for
      each cell.  Used to feed CARemap.idx directly (= sort
@@ -8735,7 +8672,7 @@ partition_addr_ki_native_object (VALUE self, CArray *ca, int axis, ca_size_t kth
   fr.axis = axis;
   fr.kth = kth;
   fr.masked_last = masked_last;
-  return ca_iter_ensure(2, fr.states, partition_addr_ki_native_object_body, (VALUE) &fr);
+  return ca_iter_ensure_computing(ca, 2, fr.states, partition_addr_ki_native_object_body, (VALUE) &fr);
 }
@@ -8826,6 +8763,10 @@ partition_addr_ki_native_fixlen (VALUE self, CArray *ca, int axis, ca_size_t kth
      positions align between iterators. */
   int8_t slab_axes[CA_RANK_MAX];
   slab_axes[0] = (int8_t) axis;
+  ca_size_t fiber_n = ca->dim[axis];
+  volatile VALUE hbuf = 0, haux = 0;
+  partition_addr_pair_fixlen *buf = ALLOCV_N(partition_addr_pair_fixlen, hbuf, fiber_n);
+  (void) haux;
   ca_iter_state st_in, st_out;
   int rc;
   rc = ca_iter_state_init_l2(&st_in,  ca, CA_SLAB_AXES,
@@ -8840,11 +8781,6 @@ partition_addr_ki_native_fixlen (VALUE self, CArray *ca, int axis, ca_size_t kth
     rb_raise(rb_eRuntimeError, "%s: output init failed rc=%d", ca_calling_method_name(), rc);
   }
-  ca_size_t fiber_n = st_in.slab_elements;
-  volatile VALUE hbuf = 0, haux = 0;
-  partition_addr_pair_fixlen *buf = ALLOCV_N(partition_addr_pair_fixlen, hbuf, fiber_n);
-  (void) haux;
   /* :view_flat semantics: kernel computes view-flat address
      (= row-major position in ca's shape, 0..elements-1) for
      each cell.  Used to feed CARemap.idx directly (= sort
@@ -9028,6 +8964,13 @@ rank_index_ki_native_i8 (VALUE self, CArray *ca, int axis, int do_stable, int de
      next_slab_axes will then be non-NULL).  Created up-front
      when input has any mask. */
   if ( ca_has_mask(ca) ) ca_create_mask(co);
+  ca_size_t fiber_n = ca->dim[axis];
+  volatile VALUE hbuf = 0, haux = 0;
+  rank_index_pair_i8 *buf = ALLOCV_N(rank_index_pair_i8, hbuf, fiber_n);
+  rank_index_pair_i8 *aux = NULL;
+  if ( do_stable ) {
+    aux = ALLOCV_N(rank_index_pair_i8, haux, fiber_n);
+  }
   ca_iter_state st_in, st_out;
   int rc;
   rc = ca_iter_state_init_l2(&st_in,  ca, CA_SLAB_AXES,
@@ -9042,14 +8985,6 @@ rank_index_ki_native_i8 (VALUE self, CArray *ca, int axis, int do_stable, int de
     rb_raise(rb_eRuntimeError, "%s: output init failed rc=%d", ca_calling_method_name(), rc);
   }
-  ca_size_t fiber_n = st_in.slab_elements;
-  volatile VALUE hbuf = 0, haux = 0;
-  rank_index_pair_i8 *buf = ALLOCV_N(rank_index_pair_i8, hbuf, fiber_n);
-  rank_index_pair_i8 *aux = NULL;
-  if ( do_stable ) {
-    aux = ALLOCV_N(rank_index_pair_i8, haux, fiber_n);
-  }
   char       *pi, *po;
   boolean8_t *mi, *mo;
   while ( ca_iter_state_next_slab_axes(&st_in,  &pi, &mi) &&
@@ -9159,6 +9094,13 @@ rank_index_ki_native_u8 (VALUE self, CArray *ca, int axis, int do_stable, int de
      next_slab_axes will then be non-NULL).  Created up-front
      when input has any mask. */
   if ( ca_has_mask(ca) ) ca_create_mask(co);
+  ca_size_t fiber_n = ca->dim[axis];
+  volatile VALUE hbuf = 0, haux = 0;
+  rank_index_pair_u8 *buf = ALLOCV_N(rank_index_pair_u8, hbuf, fiber_n);
+  rank_index_pair_u8 *aux = NULL;
+  if ( do_stable ) {
+    aux = ALLOCV_N(rank_index_pair_u8, haux, fiber_n);
+  }
   ca_iter_state st_in, st_out;
   int rc;
   rc = ca_iter_state_init_l2(&st_in,  ca, CA_SLAB_AXES,
@@ -9173,14 +9115,6 @@ rank_index_ki_native_u8 (VALUE self, CArray *ca, int axis, int do_stable, int de
     rb_raise(rb_eRuntimeError, "%s: output init failed rc=%d", ca_calling_method_name(), rc);
   }
-  ca_size_t fiber_n = st_in.slab_elements;
-  volatile VALUE hbuf = 0, haux = 0;
-  rank_index_pair_u8 *buf = ALLOCV_N(rank_index_pair_u8, hbuf, fiber_n);
-  rank_index_pair_u8 *aux = NULL;
-  if ( do_stable ) {
-    aux = ALLOCV_N(rank_index_pair_u8, haux, fiber_n);
-  }
   char       *pi, *po;
   boolean8_t *mi, *mo;
   while ( ca_iter_state_next_slab_axes(&st_in,  &pi, &mi) &&
@@ -9290,6 +9224,13 @@ rank_index_ki_native_i16 (VALUE self, CArray *ca, int axis, int do_stable, int d
      next_slab_axes will then be non-NULL).  Created up-front
      when input has any mask. */
   if ( ca_has_mask(ca) ) ca_create_mask(co);
+  ca_size_t fiber_n = ca->dim[axis];
+  volatile VALUE hbuf = 0, haux = 0;
+  rank_index_pair_i16 *buf = ALLOCV_N(rank_index_pair_i16, hbuf, fiber_n);
+  rank_index_pair_i16 *aux = NULL;
+  if ( do_stable ) {
+    aux = ALLOCV_N(rank_index_pair_i16, haux, fiber_n);
+  }
   ca_iter_state st_in, st_out;
   int rc;
   rc = ca_iter_state_init_l2(&st_in,  ca, CA_SLAB_AXES,
@@ -9304,14 +9245,6 @@ rank_index_ki_native_i16 (VALUE self, CArray *ca, int axis, int do_stable, int d
     rb_raise(rb_eRuntimeError, "%s: output init failed rc=%d", ca_calling_method_name(), rc);
   }
-  ca_size_t fiber_n = st_in.slab_elements;
-  volatile VALUE hbuf = 0, haux = 0;
-  rank_index_pair_i16 *buf = ALLOCV_N(rank_index_pair_i16, hbuf, fiber_n);
-  rank_index_pair_i16 *aux = NULL;
-  if ( do_stable ) {
-    aux = ALLOCV_N(rank_index_pair_i16, haux, fiber_n);
-  }
   char       *pi, *po;
   boolean8_t *mi, *mo;
   while ( ca_iter_state_next_slab_axes(&st_in,  &pi, &mi) &&
@@ -9421,6 +9354,13 @@ rank_index_ki_native_u16 (VALUE self, CArray *ca, int axis, int do_stable, int d
      next_slab_axes will then be non-NULL).  Created up-front
      when input has any mask. */
   if ( ca_has_mask(ca) ) ca_create_mask(co);
+  ca_size_t fiber_n = ca->dim[axis];
+  volatile VALUE hbuf = 0, haux = 0;
+  rank_index_pair_u16 *buf = ALLOCV_N(rank_index_pair_u16, hbuf, fiber_n);
+  rank_index_pair_u16 *aux = NULL;
+  if ( do_stable ) {
+    aux = ALLOCV_N(rank_index_pair_u16, haux, fiber_n);
+  }
   ca_iter_state st_in, st_out;
   int rc;
   rc = ca_iter_state_init_l2(&st_in,  ca, CA_SLAB_AXES,
@@ -9435,14 +9375,6 @@ rank_index_ki_native_u16 (VALUE self, CArray *ca, int axis, int do_stable, int d
     rb_raise(rb_eRuntimeError, "%s: output init failed rc=%d", ca_calling_method_name(), rc);
   }
-  ca_size_t fiber_n = st_in.slab_elements;
-  volatile VALUE hbuf = 0, haux = 0;
-  rank_index_pair_u16 *buf = ALLOCV_N(rank_index_pair_u16, hbuf, fiber_n);
-  rank_index_pair_u16 *aux = NULL;
-  if ( do_stable ) {
-    aux = ALLOCV_N(rank_index_pair_u16, haux, fiber_n);
-  }
   char       *pi, *po;
   boolean8_t *mi, *mo;
   while ( ca_iter_state_next_slab_axes(&st_in,  &pi, &mi) &&
@@ -9552,6 +9484,13 @@ rank_index_ki_native_i32 (VALUE self, CArray *ca, int axis, int do_stable, int d
      next_slab_axes will then be non-NULL).  Created up-front
      when input has any mask. */
   if ( ca_has_mask(ca) ) ca_create_mask(co);
+  ca_size_t fiber_n = ca->dim[axis];
+  volatile VALUE hbuf = 0, haux = 0;
+  rank_index_pair_i32 *buf = ALLOCV_N(rank_index_pair_i32, hbuf, fiber_n);
+  rank_index_pair_i32 *aux = NULL;
+  if ( do_stable ) {
+    aux = ALLOCV_N(rank_index_pair_i32, haux, fiber_n);
+  }
   ca_iter_state st_in, st_out;
   int rc;
   rc = ca_iter_state_init_l2(&st_in,  ca, CA_SLAB_AXES,
@@ -9566,14 +9505,6 @@ rank_index_ki_native_i32 (VALUE self, CArray *ca, int axis, int do_stable, int d
     rb_raise(rb_eRuntimeError, "%s: output init failed rc=%d", ca_calling_method_name(), rc);
   }
-  ca_size_t fiber_n = st_in.slab_elements;
-  volatile VALUE hbuf = 0, haux = 0;
-  rank_index_pair_i32 *buf = ALLOCV_N(rank_index_pair_i32, hbuf, fiber_n);
-  rank_index_pair_i32 *aux = NULL;
-  if ( do_stable ) {
-    aux = ALLOCV_N(rank_index_pair_i32, haux, fiber_n);
-  }
   char       *pi, *po;
   boolean8_t *mi, *mo;
   while ( ca_iter_state_next_slab_axes(&st_in,  &pi, &mi) &&
@@ -9683,6 +9614,13 @@ rank_index_ki_native_u32 (VALUE self, CArray *ca, int axis, int do_stable, int d
      next_slab_axes will then be non-NULL).  Created up-front
      when input has any mask. */
   if ( ca_has_mask(ca) ) ca_create_mask(co);
+  ca_size_t fiber_n = ca->dim[axis];
+  volatile VALUE hbuf = 0, haux = 0;
+  rank_index_pair_u32 *buf = ALLOCV_N(rank_index_pair_u32, hbuf, fiber_n);
+  rank_index_pair_u32 *aux = NULL;
+  if ( do_stable ) {
+    aux = ALLOCV_N(rank_index_pair_u32, haux, fiber_n);
+  }
   ca_iter_state st_in, st_out;
   int rc;
   rc = ca_iter_state_init_l2(&st_in,  ca, CA_SLAB_AXES,
@@ -9697,14 +9635,6 @@ rank_index_ki_native_u32 (VALUE self, CArray *ca, int axis, int do_stable, int d
     rb_raise(rb_eRuntimeError, "%s: output init failed rc=%d", ca_calling_method_name(), rc);
   }
-  ca_size_t fiber_n = st_in.slab_elements;
-  volatile VALUE hbuf = 0, haux = 0;
-  rank_index_pair_u32 *buf = ALLOCV_N(rank_index_pair_u32, hbuf, fiber_n);
-  rank_index_pair_u32 *aux = NULL;
-  if ( do_stable ) {
-    aux = ALLOCV_N(rank_index_pair_u32, haux, fiber_n);
-  }
   char       *pi, *po;
   boolean8_t *mi, *mo;
   while ( ca_iter_state_next_slab_axes(&st_in,  &pi, &mi) &&
@@ -9814,6 +9744,13 @@ rank_index_ki_native_i64 (VALUE self, CArray *ca, int axis, int do_stable, int d
      next_slab_axes will then be non-NULL).  Created up-front
      when input has any mask. */
   if ( ca_has_mask(ca) ) ca_create_mask(co);
+  ca_size_t fiber_n = ca->dim[axis];
+  volatile VALUE hbuf = 0, haux = 0;
+  rank_index_pair_i64 *buf = ALLOCV_N(rank_index_pair_i64, hbuf, fiber_n);
+  rank_index_pair_i64 *aux = NULL;
+  if ( do_stable ) {
+    aux = ALLOCV_N(rank_index_pair_i64, haux, fiber_n);
+  }
   ca_iter_state st_in, st_out;
   int rc;
   rc = ca_iter_state_init_l2(&st_in,  ca, CA_SLAB_AXES,
@@ -9828,14 +9765,6 @@ rank_index_ki_native_i64 (VALUE self, CArray *ca, int axis, int do_stable, int d
     rb_raise(rb_eRuntimeError, "%s: output init failed rc=%d", ca_calling_method_name(), rc);
   }
-  ca_size_t fiber_n = st_in.slab_elements;
-  volatile VALUE hbuf = 0, haux = 0;
-  rank_index_pair_i64 *buf = ALLOCV_N(rank_index_pair_i64, hbuf, fiber_n);
-  rank_index_pair_i64 *aux = NULL;
-  if ( do_stable ) {
-    aux = ALLOCV_N(rank_index_pair_i64, haux, fiber_n);
-  }
   char       *pi, *po;
   boolean8_t *mi, *mo;
   while ( ca_iter_state_next_slab_axes(&st_in,  &pi, &mi) &&
@@ -9945,6 +9874,13 @@ rank_index_ki_native_u64 (VALUE self, CArray *ca, int axis, int do_stable, int d
      next_slab_axes will then be non-NULL).  Created up-front
      when input has any mask. */
   if ( ca_has_mask(ca) ) ca_create_mask(co);
+  ca_size_t fiber_n = ca->dim[axis];
+  volatile VALUE hbuf = 0, haux = 0;
+  rank_index_pair_u64 *buf = ALLOCV_N(rank_index_pair_u64, hbuf, fiber_n);
+  rank_index_pair_u64 *aux = NULL;
+  if ( do_stable ) {
+    aux = ALLOCV_N(rank_index_pair_u64, haux, fiber_n);
+  }
   ca_iter_state st_in, st_out;
   int rc;
   rc = ca_iter_state_init_l2(&st_in,  ca, CA_SLAB_AXES,
@@ -9959,14 +9895,6 @@ rank_index_ki_native_u64 (VALUE self, CArray *ca, int axis, int do_stable, int d
     rb_raise(rb_eRuntimeError, "%s: output init failed rc=%d", ca_calling_method_name(), rc);
   }
-  ca_size_t fiber_n = st_in.slab_elements;
-  volatile VALUE hbuf = 0, haux = 0;
-  rank_index_pair_u64 *buf = ALLOCV_N(rank_index_pair_u64, hbuf, fiber_n);
-  rank_index_pair_u64 *aux = NULL;
-  if ( do_stable ) {
-    aux = ALLOCV_N(rank_index_pair_u64, haux, fiber_n);
-  }
   char       *pi, *po;
   boolean8_t *mi, *mo;
   while ( ca_iter_state_next_slab_axes(&st_in,  &pi, &mi) &&
@@ -10076,6 +10004,10 @@ rank_index_ki_native_f32 (VALUE self, CArray *ca, int axis, int do_stable, int d
      next_slab_axes will then be non-NULL).  Created up-front
      when input has any mask. */
   if ( ca_has_mask(ca) ) ca_create_mask(co);
+  ca_size_t fiber_n = ca->dim[axis];
+  volatile VALUE hbuf = 0, haux = 0;
+  rank_index_pair_f32 *buf = ALLOCV_N(rank_index_pair_f32, hbuf, fiber_n);
+  rank_index_pair_f32 *aux = ALLOCV_N(rank_index_pair_f32, haux, fiber_n);
   ca_iter_state st_in, st_out;
   int rc;
   rc = ca_iter_state_init_l2(&st_in,  ca, CA_SLAB_AXES,
@@ -10090,14 +10022,6 @@ rank_index_ki_native_f32 (VALUE self, CArray *ca, int axis, int do_stable, int d
     rb_raise(rb_eRuntimeError, "%s: output init failed rc=%d", ca_calling_method_name(), rc);
   }
-  ca_size_t fiber_n = st_in.slab_elements;
-  volatile VALUE hbuf = 0, haux = 0;
-  rank_index_pair_f32 *buf = ALLOCV_N(rank_index_pair_f32, hbuf, fiber_n);
-  rank_index_pair_f32 *aux = NULL;
-  if ( do_stable ) {
-    aux = ALLOCV_N(rank_index_pair_f32, haux, fiber_n);
-  }
   char       *pi, *po;
   boolean8_t *mi, *mo;
   while ( ca_iter_state_next_slab_axes(&st_in,  &pi, &mi) &&
@@ -10124,7 +10048,7 @@ rank_index_ki_native_f32 (VALUE self, CArray *ca, int axis, int do_stable, int d
     /* P.3/P.4: NaN pre-partition for float argsort -- NaN pairs are
        pushed to the tail (= NaN-at-end policy, matches sort_copy
        AC8).  Only the finite slice is sorted. */
-    ca_size_t finite_n = ca_partition_nan_pair_f32((ca_pair_f32 *) buf, (ca_size_t) k_unmasked);
+    ca_size_t finite_n = ca_partition_nan_pair_f32((ca_pair_f32 *) buf, (ca_size_t) k_unmasked, (ca_pair_f32 *) aux);
     if ( do_stable ) {
       ca_sort_merge_pair_f32((ca_pair_f32 *) buf, (ca_pair_f32 *) aux, finite_n);
     } else {
@@ -10213,6 +10137,10 @@ rank_index_ki_native_f64 (VALUE self, CArray *ca, int axis, int do_stable, int d
      next_slab_axes will then be non-NULL).  Created up-front
      when input has any mask. */
   if ( ca_has_mask(ca) ) ca_create_mask(co);
+  ca_size_t fiber_n = ca->dim[axis];
+  volatile VALUE hbuf = 0, haux = 0;
+  rank_index_pair_f64 *buf = ALLOCV_N(rank_index_pair_f64, hbuf, fiber_n);
+  rank_index_pair_f64 *aux = ALLOCV_N(rank_index_pair_f64, haux, fiber_n);
   ca_iter_state st_in, st_out;
   int rc;
   rc = ca_iter_state_init_l2(&st_in,  ca, CA_SLAB_AXES,
@@ -10227,14 +10155,6 @@ rank_index_ki_native_f64 (VALUE self, CArray *ca, int axis, int do_stable, int d
     rb_raise(rb_eRuntimeError, "%s: output init failed rc=%d", ca_calling_method_name(), rc);
   }
-  ca_size_t fiber_n = st_in.slab_elements;
-  volatile VALUE hbuf = 0, haux = 0;
-  rank_index_pair_f64 *buf = ALLOCV_N(rank_index_pair_f64, hbuf, fiber_n);
-  rank_index_pair_f64 *aux = NULL;
-  if ( do_stable ) {
-    aux = ALLOCV_N(rank_index_pair_f64, haux, fiber_n);
-  }
   char       *pi, *po;
   boolean8_t *mi, *mo;
   while ( ca_iter_state_next_slab_axes(&st_in,  &pi, &mi) &&
@@ -10261,7 +10181,7 @@ rank_index_ki_native_f64 (VALUE self, CArray *ca, int axis, int do_stable, int d
     /* P.3/P.4: NaN pre-partition for float argsort -- NaN pairs are
        pushed to the tail (= NaN-at-end policy, matches sort_copy
        AC8).  Only the finite slice is sorted. */
-    ca_size_t finite_n = ca_partition_nan_pair_f64((ca_pair_f64 *) buf, (ca_size_t) k_unmasked);
+    ca_size_t finite_n = ca_partition_nan_pair_f64((ca_pair_f64 *) buf, (ca_size_t) k_unmasked, (ca_pair_f64 *) aux);
     if ( do_stable ) {
       ca_sort_merge_pair_f64((ca_pair_f64 *) buf, (ca_pair_f64 *) aux, finite_n);
     } else {
@@ -10350,6 +10270,13 @@ rank_index_ki_native_bool (VALUE self, CArray *ca, int axis, int do_stable, int
      next_slab_axes will then be non-NULL).  Created up-front
      when input has any mask. */
   if ( ca_has_mask(ca) ) ca_create_mask(co);
+  ca_size_t fiber_n = ca->dim[axis];
+  volatile VALUE hbuf = 0, haux = 0;
+  rank_index_pair_bool *buf = ALLOCV_N(rank_index_pair_bool, hbuf, fiber_n);
+  rank_index_pair_bool *aux = NULL;
+  if ( do_stable ) {
+    aux = ALLOCV_N(rank_index_pair_bool, haux, fiber_n);
+  }
   ca_iter_state st_in, st_out;
   int rc;
   rc = ca_iter_state_init_l2(&st_in,  ca, CA_SLAB_AXES,
@@ -10364,14 +10291,6 @@ rank_index_ki_native_bool (VALUE self, CArray *ca, int axis, int do_stable, int
     rb_raise(rb_eRuntimeError, "%s: output init failed rc=%d", ca_calling_method_name(), rc);
   }
-  ca_size_t fiber_n = st_in.slab_elements;
-  volatile VALUE hbuf = 0, haux = 0;
-  rank_index_pair_bool *buf = ALLOCV_N(rank_index_pair_bool, hbuf, fiber_n);
-  rank_index_pair_bool *aux = NULL;
-  if ( do_stable ) {
-    aux = ALLOCV_N(rank_index_pair_bool, haux, fiber_n);
-  }
   char       *pi, *po;
   boolean8_t *mi, *mo;
   while ( ca_iter_state_next_slab_axes(&st_in,  &pi, &mi) &&
@@ -10476,9 +10395,9 @@ rank_index_cmp_object (const void *a, const void *b)
   return 0;
 }
-/* The object lane calls Ruby for every cell, so a walk can raise part
-   way.  Its states live in the caller's frame, and ca_iter_ensure
-   finishes them however the body leaves. */
+/* The walks can raise part way (a Ruby call per cell in the object
+   lane, an allocation in any lane).  Their states live in the caller's
+   frame, and ca_iter_ensure finishes them however the body leaves. */
 typedef struct {
   ca_iter_state states[2];
   VALUE self;
@@ -10519,6 +10438,10 @@ rank_index_ki_native_object_body (VALUE arg)
      next_slab_axes will then be non-NULL).  Created up-front
      when input has any mask. */
   if ( ca_has_mask(ca) ) ca_create_mask(co);
+  ca_size_t fiber_n = ca->dim[axis];
+  volatile VALUE hbuf = 0, haux = 0;
+  rank_index_pair_object *buf = ALLOCV_N(rank_index_pair_object, hbuf, fiber_n);
+  rank_index_pair_object *aux = ALLOCV_N(rank_index_pair_object, haux, fiber_n);
   int rc;
   rc = ca_iter_state_init_l2(&st_in,  ca, CA_SLAB_AXES,
                              slab_axes, 1, 0);
@@ -10532,11 +10455,6 @@ rank_index_ki_native_object_body (VALUE arg)
     rb_raise(rb_eRuntimeError, "%s: output init failed rc=%d", ca_calling_method_name(), rc);
   }
-  ca_size_t fiber_n = st_in.slab_elements;
-  volatile VALUE hbuf = 0, haux = 0;
-  rank_index_pair_object *buf = ALLOCV_N(rank_index_pair_object, hbuf, fiber_n);
-  rank_index_pair_object *aux = ALLOCV_N(rank_index_pair_object, haux, fiber_n);
   char       *pi, *po;
   boolean8_t *mi, *mo;
   while ( ca_iter_state_next_slab_axes(&st_in,  &pi, &mi) &&
@@ -10633,7 +10551,7 @@ rank_index_ki_native_object (VALUE self, CArray *ca, int axis, int do_stable, in
   fr.axis = axis;
   fr.do_stable = do_stable;
   fr.dense = dense;
-  return ca_iter_ensure(2, fr.states, rank_index_ki_native_object_body, (VALUE) &fr);
+  return ca_iter_ensure_computing(ca, 2, fr.states, rank_index_ki_native_object_body, (VALUE) &fr);
 }
@@ -10674,6 +10592,10 @@ rank_index_ki_native_fixlen (VALUE self, CArray *ca, int axis, int do_stable, in
      next_slab_axes will then be non-NULL).  Created up-front
      when input has any mask. */
   if ( ca_has_mask(ca) ) ca_create_mask(co);
+  ca_size_t fiber_n = ca->dim[axis];
+  volatile VALUE hbuf = 0, haux = 0;
+  rank_index_pair_fixlen *buf = ALLOCV_N(rank_index_pair_fixlen, hbuf, fiber_n);
+  (void) haux;
   ca_iter_state st_in, st_out;
   int rc;
   rc = ca_iter_state_init_l2(&st_in,  ca, CA_SLAB_AXES,
@@ -10688,11 +10610,6 @@ rank_index_ki_native_fixlen (VALUE self, CArray *ca, int axis, int do_stable, in
     rb_raise(rb_eRuntimeError, "%s: output init failed rc=%d", ca_calling_method_name(), rc);
   }
-  ca_size_t fiber_n = st_in.slab_elements;
-  volatile VALUE hbuf = 0, haux = 0;
-  rank_index_pair_fixlen *buf = ALLOCV_N(rank_index_pair_fixlen, hbuf, fiber_n);
-  (void) haux;
   char       *pi, *po;
   boolean8_t *mi, *mo;
   while ( ca_iter_state_next_slab_axes(&st_in,  &pi, &mi) &&