Skip to content

[autotuner] Add benchmark_batch as unified entry point for autotuner benchmarking#1810

Merged
jansel merged 6 commits into
pytorch:mainfrom
hinriksnaer:autotuner-cleanup
Mar 26, 2026
Merged

[autotuner] Add benchmark_batch as unified entry point for autotuner benchmarking#1810
jansel merged 6 commits into
pytorch:mainfrom
hinriksnaer:autotuner-cleanup

Conversation

@hinriksnaer
Copy link
Copy Markdown
Collaborator

Motivation

The base_search.py file is both very frequently modified and getting bigger and bigger by the day. This is a first set of changes intended to both standardize some of the steps and attempt to make things more isolated and decoupled.

for this first one, all search algorithms need to compile and benchmark configs, but they entered the pipeline through different methods (parallel_benchmark, parallel_benchmark_population, parallel_benchmark_flat). There was no single public entry point, and metrics were updated as side effects inside benchmark_function, making it hard to reason about where state changes happen.

I intend to make a couple of these as long as we agree it's beneficial, but for now I want to keep these somewhat minimal and incremental. Hopefully this will make expanding autotuning capabilities less error prone and simple in the future.

@meta-cla meta-cla Bot added the CLA Signed This label is managed by the Meta Open Source bot. label Mar 24, 2026
Comment thread helion/autotuner/base_search.py Outdated
@fulvius31 fulvius31 requested a review from jansel March 25, 2026 14:32
@jansel jansel merged commit 15fd389 into pytorch:main Mar 26, 2026
18 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

CLA Signed This label is managed by the Meta Open Source bot.

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants