[Mlir-commits] [mlir] [MLIR][mesh] Mesh fixes (PR #124724)

Renato Golin llvmlistbot at llvm.org
Tue Feb 11 09:18:18 PST 2025


================
@@ -164,6 +164,16 @@ func.func @mesh_shard_shape() {
   return
 }
 
+// CHECK-LABEL: func @mesh_get_sharding
+// CHECK-SAME: %[[ARG:.*]]: tensor<4x8xf32>
+func.func @mesh_get_sharding(%arg0 : tensor<4x8xf32>) -> !mesh.sharding {
+  // CHECK-NEXT: %[[S:.*]] = mesh.sharding @mesh1 split_axes = {{\[\[}}], [0]] : !mesh.sharding
+  %s = mesh.sharding @mesh1 split_axes = [[], [0]] : !mesh.sharding
----------------
rengolin wrote:

`%s` is unused and unrelated, why do you need it in this test?

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


More information about the Mlir-commits mailing list