Skip to content

Enable the use of layered CF output for checkpoint-restart - #468

Open
Ed Hone (EdHone) wants to merge 13 commits into
MetOffice:mainfrom
EdHone:ugrid-ckp-flag
Open

Ed Hone (EdHone) wants to merge 13 commits into
MetOffice:mainfrom
EdHone:ugrid-ckp-flag

Conversation

@EdHone

@EdHone Ed Hone (EdHone) commented Sep 15, 2026 •

Copy link
Copy Markdown
Contributor

PR Summary

Sci/Tech Reviewer: Mike Hobson (@mike-hobson)
Code Reviewer: svadams (@svadams)

The checkpointing I/O system used by some applications in lfric_apps uses 'legacy' checkpointing domains which are extremely non-performance at scale. The aim is convert to using the same UGRID format for output as the diagnostic files, which would involve changing the way that fields on the W2 function space are set up for I/O in XIOS. However, progress has been difficult for implementing this goal, as the UGRID implementation inside XIOS has some requirements and bugs which we cannot circumvent at the required timescales.
To get a performant checkpointing system working as soon as possible, this PR enables these applications to use checkpoint files to use a layered CF I/O implementation, which will circumvent the bugs in UGRID whilst maintaining the improved performance gains.

Code Quality Checklist

  • I have performed a self-review of my own code
  • My code follows the project's style guidelines
  • Comments have been included that aid understanding and enhance the readability of the code
  • My changes generate no new warnings
  • All automated checks in the CI pipeline have completed successfully

Testing

  • I have tested this change locally, using the LFRic Core rose-stem suite
  • If required (e.g. API changes) I have also run the LFRic Apps test suite using this branch
  • If any tests fail (rose-stem or CI) the reason is understood and acceptable (e.g. kgo changes)
  • I have added tests to cover new functionality as appropriate (e.g. system tests, unit tests, etc.)
  • Any new tests have been assigned an appropriate amount of compute resource and have been allocated to an appropriate testing group (i.e. the developer tests are for jobs which use a small amount of compute resource and complete in a matter of minutes)

trac.log

Test Suite Results - lfric_core - lfric_core-ugrid-ckp-flag/run3

Suite Information

Item Value
Suite Name lfric_core-ugrid-ckp-flag/run3
Suite User edward.hone
Workflow Start 2026-09-30T14:22:35
Groups Run developer
Dependency Reference Main Like
lfric_core EdHone/lfric_core@ugrid-ckp-flag False
SimSys_Scripts MetOffice/SimSys_Scripts@cab3315 True

Task Information

✅ succeeded tasks - 433

Security Considerations

  • I have reviewed my changes for potential security issues
  • Sensitive data is properly handled (if applicable)
  • Authentication and authorisation are properly implemented (if applicable)

Performance Impact

  • Performance of the code has been considered and, if applicable, suitable performance measurements have been conducted

AI Assistance and Attribution

  • Some of the content of this change has been produced with the assistance of Generative AI tool name (e.g., Met Office Github Copilot Enterprise, Github Copilot Personal, ChatGPT GPT-4, etc) and I have followed the Simulation Systems AI policy (including attribution labels)

Documentation

  • Where appropriate I have updated documentation related to this change and confirmed that it builds correctly

PSyclone Approval

  • If you have edited any PSyclone-related code (e.g. PSyKAl-lite, Kernel interface, optimisation scripts, LFRic data structure code) then please contact the HPC Optimisation Team

Sci/Tech Review

  • I understand this area of code and the changes being added
  • The proposed changes correspond to the pull request description
  • Documentation is sufficient (do documentation papers need updating)
  • Sufficient testing has been completed

(Please alert the code reviewer via a tag when you have approved the SR)

Code Review

  • All dependencies have been resolved
  • Related Issues have been properly linked and addressed
  • CLA compliance has been confirmed
  • Code quality standards have been met
  • Tests are adequate and have passed
  • Documentation is complete and accurate
  • Security considerations have been addressed
  • Performance impact is acceptable

@mike-hobson Mike Hobson (mike-hobson) left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

This change looks fine - just a couple of very minor points. Obviously, I've looked at the linked ticket for context. I'm a little concerned about the way we are controling legacy mode. Currently, legacy mode is selected if the run has a planar, biperiodic mesh. This allows the SCM runs to use legacy mode.

When we move to XIOS 3, the SCM checkpointing should work in layered mode. We will then want to remove the if-tests that automaticaaly select legacy mode. At that point, we will no longer be able to select legacy mode, so we'll have dead code. To avoid getting into this situation, I think we have a couple of options:

  1. Given that XIOS 3 is very close, we could remove the support for legacy mode, in this change. That would mean this would have to wait until after XIOS 3 is done, but putting all this code in, so it can be removed in weeks seems wasteful and prone to error.
  2. Introduce a namelist switch to turn legacy checkpointing on, then set the switch to "on" for the SCM jobs. That way, when the SCM runs work with layered checkpointing, it would be easy to switch them, and if we ever wanted to make any run use legacy checkpointing in the future, that would also be really easy.

I probably edge towards the second option. I know it wouldn't be particulalrly easy, as you have to pass the switch variable through the interface to the procedure pointer, but in my many experiments into checkpointing, I have already done this. If you'd like to see what I did:
https://github.com/mike-hobson/lfric_apps/tree/apps_layered
and
https://github.com/mike-hobson/lfric_core/tree/core_layered

What do you think?

module lfric_xios_metafile_mod

use constants_mod, only: str_def, i_def, l_def
use constants_mod, only: str_def, i_def, l_def, imdi

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Looks like you never implemented whatever you added imdi for. I think it can just be removed.

Comment on lines +380 to +391
! Check if file convention has been defined in iodef.xml config
call xios_is_defined_file_attr(self%xios_id, convention=convention_defined)
if (convention_defined) then
call xios_get_file_attr(self%xios_id, convention=iodef_file_convention)
if (trim(iodef_file_convention) == "CF") then
self%file_convention = CONVENTION_CF
else if (trim(iodef_file_convention) == "UGRID") then
self%file_convention = CONVENTION_UGRID
end if
else if (self%file_convention == undef_file_convention) then
self%file_convention = CONVENTION_UGRID
end if

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I realise that it's unlikely that these if-tests wouldn't catch all cases, but it would still be good to have an else option to catch any potential errors, rather than let the code continue without setting the file convention.

This branch has not been deployed

No deployments
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