Skip to content

perf(training): increase default compilation - #1317

Draft
cathalobrien wants to merge 8 commits into
perf/compile-changesfrom
perf/increase-default-compilation
Draft

perf(training): increase default compilation#1317
cathalobrien wants to merge 8 commits into
perf/compile-changesfrom
perf/increase-default-compilation

Conversation

@cathalobrien

@cathalobrien cathalobrien commented Aug 6, 2026

Copy link
Copy Markdown
Contributor

Description

Following on the rest of this stack, this PR changes the default compilation.

Draft PR while I benchmark different setups to determine a reasonable default config.

I refactored the model yamls and pulled the list of compiled modules into their own files, so you can easily test different levels of compilation

anemoi-training train --config-name=config_n320_gt model/compile=none
anemoi-training train --config-name=config_n320_gt model/compile=medium
anemoi-training train --config-name=config_n320_gt model/compile=max

(it is not a config breaking PR, I followed the existing syntax but refactored into their own files)

currently compile/max.yaml compiles the entire mapper and processor block. compile/medium.yaml maintains the defaults before this PR.

By opening this pull request, I affirm that all authors agree to the Contributor License Agreement.

@github-project-automation github-project-automation Bot moved this to To be triaged in Anemoi-dev Aug 6, 2026
@cathalobrien cathalobrien changed the title perf(training): compile mappers and processors by default perf(training): increase default compilation Aug 6, 2026
@cathalobrien cathalobrien added the ATS Approval Not Needed No approval needed by ATS label Aug 6, 2026
@cathalobrien
cathalobrien force-pushed the perf/increase-default-compilation branch from e7bf043 to 363a6c2 Compare August 6, 2026 13:08
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

ATS Approval Not Needed No approval needed by ATS training

Projects

Status: To be triaged

Development

Successfully merging this pull request may close these issues.

1 participant