[Mlir-commits] [mlir] [llvm] [mlir][mesh] Add spmdization pass (PR #80518)

Boian Petkantchin llvmlistbot at llvm.org
Mon Feb 5 14:07:46 PST 2024


================
@@ -0,0 +1,131 @@
+// RUN: mlir-opt --mesh-spmdization --test-constant-fold %s | FileCheck %s
----------------
sogartar wrote:

`mesh.local_split` would a useful op to abstract the above blob of ops. I will add it in subsequent PR and make resharding produce it instead. Then have another rewrite pattern to lower it to the slicing logic.

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


More information about the Mlir-commits mailing list