count group benchmark check

Author: coderfender

datafusion/functions-aggregate/benches/count_distinct.rs

@@ -224,6 +224,7 @@ fn count_distinct_groups_benchmark(c: &mut Criterion) {
         let (_schema, args) = prepare_args(DataType::Int64);
 
         if count_fn.groups_accumulator_supported(args.clone()) {
+<<<<<<< Updated upstream
             c.bench_function(&format!("count_distinct_groups {name}"), |b| {
                 b.iter(|| {
                     let (_schema, args) = prepare_args(DataType::Int64);
@@ -238,6 +239,25 @@ fn count_distinct_groups_benchmark(c: &mut Criterion) {
                     acc.evaluate(EmitTo::All).unwrap()
                 })
             });
+=======
+            c.bench_function(
+                &format!("count_distinct_groups i64 {num_groups} groups"),
+                |b| {
+                    b.iter(|| {
+                        let mut acc =
+                            count_fn.create_groups_accumulator(args.clone()).unwrap();
+                        acc.update_batch(
+                            std::slice::from_ref(&values),
+                            &group_indices,
+                            None,
+                            num_groups,
+                        )
+                        .unwrap();
+                        acc.evaluate(EmitTo::All).unwrap()
+                    })
+                },
+            );
+>>>>>>> Stashed changes
         } else {
             c.bench_function(&format!("count_distinct_groups {name}"), |b| {
                 b.iter(|| {