YABLoCo: Yet Another Benchmark for Long Context Code Generation

TL;DR

This paper introduces YABLoCo, a comprehensive benchmark for evaluating large language models on long-context code generation tasks in large C and C++ repositories, addressing a significant gap in existing benchmarks.

Contribution

The paper presents a new benchmark with large repositories, including a scalable evaluation pipeline and tools for visual analysis, specifically targeting long-context code generation in C and C++.

Findings

Benchmark includes 215 functions from repositories with up to 2 million lines of code.

Introduces a scalable evaluation pipeline for efficient metric computation.

Provides tools for visual analysis of generated code.

Abstract

Large Language Models demonstrate the ability to solve various programming tasks, including code generation. Typically, the performance of LLMs is measured on benchmarks with small or medium-sized context windows of thousands of lines of code. At the same time, in real-world software projects, repositories can span up to millions of LoC. This paper closes this gap by contributing to the long context code generation benchmark (YABLoCo). The benchmark featured a test set of 215 functions selected from four large repositories with thousands of functions. The dataset contained metadata of functions, contexts of the functions with different levels of dependencies, docstrings, functions bodies, and call graphs for each repository. This paper presents three key aspects of the contribution. First, the benchmark aims at function body generation in large repositories in C and C++, two languages not covered by previous benchmarks. Second, the benchmark contains large repositories from 200K to 2,000K LoC. Third, we contribute a scalable evaluation pipeline for efficient computing of the target metrics and a tool for visual analysis of generated code. Overall, these three aspects allow for evaluating code generation in large repositories in C and C++.