(arrow-rs) branch master updated: add filter_kernel benchmark for run array (#6706)

[dheres]

This is an automated email from the ASF dual-hosted git repository.

dheres pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/arrow-rs.git


The following commit(s) were added to refs/heads/master by this push:
     new 28878d375c add filter_kernel benchmark for run array (#6706)
28878d375c is described below

commit 28878d375c5bb1c2f5adb46b287e16cea768b6e7
Author: delamarch3 <68732277+delamar...@users.noreply.github.com>
AuthorDate: Sun Nov 10 18:24:04 2024 +0000

    add filter_kernel benchmark for run array (#6706)
---
 arrow/benches/filter_kernels.rs | 13 ++++++++++++-
 1 file changed, 12 insertions(+), 1 deletion(-)

diff --git a/arrow/benches/filter_kernels.rs b/arrow/benches/filter_kernels.rs
index ae5eec386e..18b6e2a938 100644
--- a/arrow/benches/filter_kernels.rs
+++ b/arrow/benches/filter_kernels.rs
@@ -23,7 +23,7 @@ use arrow::util::bench_util::*;
 
 use arrow::array::*;
 use arrow::compute::filter;
-use arrow::datatypes::{Field, Float32Type, Int32Type, Schema, UInt8Type};
+use arrow::datatypes::{Field, Float32Type, Int32Type, Int64Type, Schema, 
UInt8Type};
 
 use arrow_array::types::Decimal128Type;
 use criterion::{criterion_group, criterion_main, Criterion};
@@ -283,6 +283,17 @@ fn add_benchmark(c: &mut Criterion) {
         "filter context mixed string view low selectivity (kept 1/1024)",
         |b| b.iter(|| bench_built_filter(&sparse_filter, &data_array)),
     );
+
+    let data_array = create_primitive_run_array::<Int32Type, Int64Type>(size, 
size);
+    c.bench_function("filter run array (kept 1/2)", |b| {
+        b.iter(|| bench_built_filter(&filter, &data_array))
+    });
+    c.bench_function("filter run array high selectivity (kept 1023/1024)", |b| 
{
+        b.iter(|| bench_built_filter(&dense_filter, &data_array))
+    });
+    c.bench_function("filter run array low selectivity (kept 1/1024)", |b| {
+        b.iter(|| bench_built_filter(&sparse_filter, &data_array))
+    });
 }
 
 criterion_group!(benches, add_benchmark);