zackc6 wrote: > What about the inverse case `all_gather(all_slice())`? It'd be nice to have that as as well. @fschlimb Thanks a lot for the suggestion, I added the case too. If the change is ok, could you help me to merge ? :) https://github.com/llvm/llvm-project/pull/193906