Function BM_CastDispatchBaseline

cpp/src/arrow/compute/function_benchmark.cc:67–98

}

void BM_CastDispatchBaseline(benchmark::State& state) {
  // Repeatedly invoke a trivial Cast with all dispatch outside the hot loop
  random::RandomArrayGenerator rag(kSeed);

  auto int_array = rag.Int64(1, 0, 1 << 20);
  auto double_type = float64();
  CastOptions cast_options;
  cast_options.to_type = double_type;
  ASSERT_OK_AND_ASSIGN(auto cast_function, internal::GetCastFunction(*double_type));
  ASSERT_OK_AND_ASSIGN(auto cast_kernel,
                       cast_function->DispatchExact({int_array->type()}));
  const auto& exec = static_cast<const ScalarKernel*>(cast_kernel)->exec;

  ExecContext exec_context;
  KernelContext kernel_context(&exec_context);
  auto cast_state = cast_kernel
                        ->init(&kernel_context,
                               KernelInitArgs{cast_kernel, {double_type}, &cast_options})
                        .ValueOrDie();
  kernel_context.SetState(cast_state.get());

  ExecSpan input({ExecValue(*int_array->data())}, 1);
  ExecResult result;
  ASSERT_OK_AND_ASSIGN(std::shared_ptr<Array> result_space,
                       MakeArrayOfNull(double_type, 1));
  result.array_span_mutable()->SetMembers(*result_space->data());
  for (auto _ : state) {
    ABORT_NOT_OK(exec(&kernel_context, input, &result));
  }

  state.SetItemsProcessed(state.iterations() * kScalarCount);
}

void BM_AddDispatch(benchmark::State& state) {
  ExecContext exec_context;