order must be uppercase
@@ -265,13 +265,13 @@ cpdef inline tuple _can_use_cub_block_reduction(
# check reduction axes, if not contiguous then fall back to old kernel
if in_arr._f_contiguous:
- order = 'f'
+ order = 'F'
if not cub._cub_device_segmented_reduce_axis_compatible(
reduce_axis, in_arr.ndim, order):
return None
axis_permutes_cub = tuple(sorted(reduce_axis) + sorted(out_axis))
elif in_arr._c_contiguous:
- order = 'c'
+ order = 'C'
if not cub._cub_device_segmented_reduce_axis_compatible(
reduce_axis, in_arr.ndim, order):
return None
