[Mlir-commits] [mlir] [mlir][shard] Add collective simplify rewrites (PR #193982)

[llvmlistbot at llvm.org]

zackc6 wrote:

> Good to see your contribution! Do you actually see such cases in your workloads?

Hi @fschlimb, yes 

For the reduceScatter + allGather pattern is a little tricky, we have a cost model determine when it is good to simplify (eg. low latency scenario with small input), might not suitable for all the general usages.

So I created another PR instead,
https://github.com/llvm/llvm-project/pull/193906

forget to close this one, sorry for the inconveniences caused

https://github.com/llvm/llvm-project/pull/193982