https://github.com/MaheshRavishankar requested changes to this pull request. Thanks for the change. We have been doing something similar downstream as well. @yzhang93 can you help review this and guide it based on the use case we have worked through. https://github.com/llvm/llvm-project/pull/89782