https://github.com/Artem-B commented: LGTM in principle. Would you be willing to land it as a separate commit? It's a nice self-contained improvement that's not directly related to the cluster_dim changes. https://github.com/llvm/llvm-project/pull/109548