Function BM_Comprehension_Trace

eval/tests/modern_benchmark_test.cc:590–619

BENCHMARK(BM_Comprehension)->Range(1, 1 << 20);

void BM_Comprehension_Trace(benchmark::State& state) {
  RuntimeOptions options = GetOptions();
  options.enable_recursive_tracing = true;

  options.comprehension_max_iterations = 10000000;
  auto runtime = StandardRuntimeOrDie(options);
  google::protobuf::Arena arena;
  Expr expr;
  Activation activation;
  ASSERT_TRUE(google::protobuf::TextFormat::ParseFromString(kListSum, &expr));

  ASSERT_OK_AND_ASSIGN(auto cel_expr,
                       ProtobufRuntimeAdapter::CreateProgram(*runtime, expr));

  auto list_builder = cel::NewListValueBuilder(&arena);

  int len = state.range(0);
  list_builder->Reserve(len);
  for (int i = 0; i < len; i++) {
    ASSERT_THAT(list_builder->Add(IntValue(1)), IsOk());
  }
  activation.InsertOrAssignValue("list_var", std::move(*list_builder).Build());

  for (auto _ : state) {
    ASSERT_OK_AND_ASSIGN(cel::Value result,
                         cel_expr->Trace(&arena, activation, EmptyCallback));
    ASSERT_TRUE(InstanceOf<IntValue>(result));
    ASSERT_EQ(Cast<IntValue>(result), len);
  }
}

BENCHMARK(BM_Comprehension_Trace)->Range(1, 1 << 20);