NVIDIA / NVIDIA/compute-eval

compute-eval: correctness-focused benchmark, not performance-oriented?

Open
#7 3 comments 0 reactions 0 assignees View on GitHub

Nobody has claimed this yet.

Dominant language
Python
Stars
149
Forks
18
PR merge metrics
No merged PRs in 30d

Description

Hi, thanks for releasing compute-eval, it’s a very useful tool.

After reviewing the codebase, it seems that the benchmark primarily measures end-to-end time from compilation to execution, without reporting the actual CUDA kernel runtime. Moreover, the evaluation logic centers on pass@k correctness.

This makes compute-eval appear to be mainly a correctness benchmark: assessing whether models can generate valid CUDA kernels, rather than a performance-oriented benchmark like KernelBench or robust-kbench, which evaluate kernel execution speed and optimization quality.

Could you clarify if this correctness-first design is intentional? Are there any plans to incorporate performance metrics in future releases, so that it can also benchmark kernel optimization capabilities?

Contributor guide

Open the contributing guide

First steps

  1. Read the whole issue, then the project's contributing guide.
  2. Comment on the issue to say you are picking it up — it saves two people doing the same work.
  3. Fork the repository and make your change on a branch.
  4. Open a pull request that references the issue number.

Research direction

Start by reading compute_eval/execution.py around the end-to-end timing and compute_eval/evaluation.py around pass@k, then compare the benchmark scope described in the issue. Done would require a maintainer decision on whether performance metrics belong in the project and a defined, agreed scope for any follow-up work.

Written by the indexing model from the issue text.

Assessment

Tech stack
python
Domain
machine-learning, performance, testing-qa
Issue type
Feature
Difficulty
5/5
Estimated time
Over a week
Activity status
Stale
Clarity
Needs clarification
Newbie friendliness
20/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.