https://github.com/david-arm approved this pull request. LGTM, but can you address the nit for the tests before landing? Also, I think it would be good to change the commit message to reflect what you've done in the patch. https://github.com/llvm/llvm-project/pull/106699