Function BM_NestedComprehension_Trace

eval/tests/modern_benchmark_test.cc:981–1014

BENCHMARK(BM_NestedComprehension)->Range(1, 1 << 10);

void BM_NestedComprehension_Trace(benchmark::State& state) {
  Expr expr;
  ASSERT_TRUE(google::protobuf::TextFormat::ParseFromString(kNestedListSum, &expr));

  RuntimeOptions options = GetOptions();
  options.comprehension_max_iterations = 10000000;
  options.enable_comprehension_list_append = true;
  options.enable_recursive_tracing = true;

  auto runtime = StandardRuntimeOrDie(options);

  google::protobuf::Arena arena;
  Activation activation;

  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());

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

  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 * len);
  }
}

BENCHMARK(BM_NestedComprehension_Trace)->Range(1, 1 << 10);