Skip to content

[Script] Simple script to run kernel bench modules - #214

Merged
rengolin merged 2 commits into
mainfrom
kb_scripts
Jul 7, 2026
Merged

[Script] Simple script to run kernel bench modules#214
rengolin merged 2 commits into
mainfrom
kb_scripts

Conversation

@rengolin

@rengolin rengolin commented Jul 6, 2026

Copy link
Copy Markdown
Member

Batched script to execute a number of kernels and capture a lot of information on the execution for every kernel:

  • kernel-bench command line, as driven by test-kernel-bench.py
  • Full MLIR pipeline dump, into a log file
  • Object and assembly files
  • Validation against PyTorch, benchmark numbers, etc.

The main point of this script is to help remote execution in a batched way, for example when SSH/SRUN on remote compute nodes, or multiple nodes, for wide collection of data and benchmark results.

The secondary point is to help debug individual compilations by providing as much information as possible for various choices (different cmd-line args) and allowing one to compare them offline.

Also, update kernel-bench's ability to receive names for the object and assembly files directly, to make it easier to automate and not pollute the repo or the KernelBench submodule.

Also, update kernel-bench's ability to receive names for the object and
assembly files directly, to make it easier to automate and not pollute
the repo or the KernelBench submodule.
@rengolin
rengolin requested a review from adam-smnk July 6, 2026 12:59
Comment thread tools/kernel-bench Outdated
Comment thread scripts/KernelBench/run_kernels.sh
@rengolin
rengolin merged commit cd5abe1 into main Jul 7, 2026
4 checks passed
@rengolin
rengolin deleted the kb_scripts branch July 7, 2026 09:13
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.

3 participants