https://github.com/Artem-B approved this pull request. Few nits. LGTM otherwise. I'm actually surprised that there are no changes at all for the tests. I suspect that we may not have enough coverage. https://github.com/llvm/llvm-project/pull/137666