[Mlir-commits] [mlir] [mlir][mesh] Add endomorphism simplification for all-reduce (PR #73150)

Chengji Yao llvmlistbot at llvm.org
Thu Dec 7 22:14:14 PST 2023


================
@@ -0,0 +1,131 @@
+// RUN: mlir-opt -test-mesh-simplifications %s | FileCheck %s
----------------
yaochengji wrote:

Yes, it's a complex problem. A common approach is to leave it unsimplified. XLA has [a pass](https://github.com/openxla/xla/blob/main/xla/service/all_reduce_reassociate.h) with similar logic, maybe we can check its logic there 

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


More information about the Mlir-commits mailing list