[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