From 2000f6c77588ee765018b5f6c87d8623e08b0056 Mon Sep 17 00:00:00 2001 From: Tom Natan Date: Wed, 28 Aug 2024 07:13:09 -0700 Subject: [PATCH] failing test test PiperOrigin-RevId: 668438132 --- .../transforms/propagation/test/op_sharding_rule_registry.mlir | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/shardy/dialect/sdy/transforms/propagation/test/op_sharding_rule_registry.mlir b/shardy/dialect/sdy/transforms/propagation/test/op_sharding_rule_registry.mlir index 0e867bd..4957f9d 100644 --- a/shardy/dialect/sdy/transforms/propagation/test/op_sharding_rule_registry.mlir +++ b/shardy/dialect/sdy/transforms/propagation/test/op_sharding_rule_registry.mlir @@ -2,7 +2,7 @@ // CHECK-LABEL: func @pointwise_op func.func @pointwise_op(%arg0: tensor<2x1x4xf32>) -> tensor<2x1x4xf32> { - // CHECK: sdy.sharding_rule = #sdy.op_sharding_rule<([i, j, k], [i, j, k])->([i, j, k]) {i=2, j=1, k=4}> + // CHECK: sdy.sharding_rule = #sdy.op_sharding_rule<([i, j, k], [i, j, k])->([i, j, k]) {i=3, j=1, k=4}> %0 = stablehlo.add %arg0, %arg0: tensor<2x1x4xf32> return %0 : tensor<2x1x4xf32> }