Skip to content

Forward TPC-H scale factor to benchmark pytest runner - #361

Open
pentschev wants to merge 1 commit into
rapidsai:mainfrom
pentschev:fix-nvl72-scale-factor
Open

Forward TPC-H scale factor to benchmark pytest runner#361
pentschev wants to merge 1 commit into
rapidsai:mainfrom
pentschev:fix-nvl72-scale-factor

Conversation

@pentschev

Copy link
Copy Markdown
Member

This fixes NVL72 Presto benchmark runs where launch-run.sh -s <SF> correctly sets the schema/launcher scale factor, but run_benchmark.sh does not forward that value to pytest.

Without this, the TPC-H query fixture falls back to reading metadata.json next to the mounted dataset and can fail with KeyError: 'scale_factor' before any query runs.

Changes:

  • Add --scale-factor support to presto/scripts/run_benchmark.sh.
  • Pass the SLURM launcher scale factor through presto/slurm/presto-nvl72/functions.sh.
  • Make get_scale_factor_from_file() accept both top-level scale_factor and nested options.scale_factor.

This fixes NVL72 Presto benchmark runs where `launch-run.sh -s <SF>`
correctly sets the schema/launcher scale factor, but `run_benchmark.sh`
does not forward that value to pytest.

Without this, the TPC-H query fixture falls back to reading
`metadata.json` next to the mounted dataset and can fail with `KeyError:
'scale_factor'` before any query runs.

Changes:
- Add `--scale-factor` support to `presto/scripts/run_benchmark.sh`.
- Pass the SLURM launcher scale factor through
`presto/slurm/presto-nvl72/functions.sh`.
- Make `get_scale_factor_from_file()` accept both top-level
`scale_factor` and nested `options.scale_factor`.
@pentschev
pentschev requested a review from a team as a code owner June 16, 2026 18:37
@pentschev
pentschev requested a review from karthikeyann June 16, 2026 18:37
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant