https://github.com/arichardson commented: This looks good to me. Since this uses some of my code could you please add `Co-authored-by: Alex Richardson <alexrichardson at google.com>` to the commit message. https://github.com/llvm/llvm-project/pull/112007