[Mlir-commits] [mlir] [mlir][mesh] Insert resharding during sharding propagation (PR #84514)

Chengji Yao llvmlistbot at llvm.org
Sun Mar 10 20:40:36 PDT 2024


================
@@ -0,0 +1,34 @@
+// RUN: mlir-opt \
+// RUN:   --verify-each \
+// RUN:   --pass-pipeline="builtin.module(func.func(sharding-propagation))" \
+// RUN:   %s | FileCheck %s
+
+mesh.mesh @mesh_2_2(shape = 2)
+
+// CHECK-LABEL: func @matmul_shard_prallel_axis
+func.func @matmul_shard_prallel_axis(
----------------
yaochengji wrote:

```suggestion
func.func @matmul_shard_parallel_axis(
```

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


More information about the Mlir-commits mailing list