Function bench_array_sort

datafusion/functions-nested/benches/array_sort.rs:134–192

Vary elements_per_row over [5, 20, 100, 1000]: for small arrays, per-row overhead dominates, whereas for larger arrays the sort kernel dominates.

(c: &mut Criterion)

/// Vary elements_per_row over [5, 20, 100, 1000]: for small arrays, per-row
/// overhead dominates, whereas for larger arrays the sort kernel dominates.
fn bench_array_sort(c: &mut Criterion) {
    let mut group = c.benchmark_group("array_sort");
    let udf = ArraySort::new();

    // Int32 arrays
    for &elements_per_row in &[5, 20, 100, 1000] {
        let array = create_int32_list_array(NUM_ROWS, elements_per_row, false);
        group.bench_with_input(
            BenchmarkId::new("int32", elements_per_row),
            &elements_per_row,
            |b, _| {
                b.iter(|| {
                    black_box(invoke_array_sort(&udf, &array));
                });
            },
        );
    }

    // Int32 with nulls in the outer list (10% null rows), single size
    {
        let array = create_int32_list_array(NUM_ROWS, 50, true);
        group.bench_function("int32_with_nulls", |b| {
            b.iter(|| {
                black_box(invoke_array_sort(&udf, &array));
            });
        });
    }

    // Int32 with null elements (~10% of elements within rows are null)
    for &elements_per_row in &[5, 20, 100, 1000] {
        let array =
            create_int32_list_array_with_null_elements(NUM_ROWS, elements_per_row);
        group.bench_with_input(
            BenchmarkId::new("int32_null_elements", elements_per_row),
            &elements_per_row,
            |b, _| {
                b.iter(|| {
                    black_box(invoke_array_sort(&udf, &array));
                });
            },
        );
    }

    // String arrays
    for &elements_per_row in &[5, 20, 100, 1000] {
        let array = create_string_list_array(NUM_ROWS, elements_per_row);
        group.bench_with_input(
            BenchmarkId::new("string", elements_per_row),
            &elements_per_row,
            |b, _| {
                b.iter(|| {
                    black_box(invoke_array_sort(&udf, &array));
                });
            },
        );
    }

    group.finish();