https://github.com/nicolasvasilache approved this pull request. Looks good, thanks much! Can you please fix the formatting before landing? I see an error message from some presubmit. https://github.com/llvm/llvm-project/pull/66930