[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