[mlir][shard, mpi] Allow more than one last axis to be "unsplit"#180754

fschlimb merged 3 commits into

llvm:mainllvm/llvm-project:mainfrom

fschlimb:shard-unsplit-multfschlimb/llvm-project:shard-unsplit-multCopy head branch name to clipboard