https://github.com/WenleiHe commented: Haven't looked closely. But this does feel cleaner than previous implementation. Thanks for working through this. https://github.com/llvm/llvm-project/pull/95135