failing test test

tomnatan30 · copybara-github · commit 2000f6c77588 · 2024-08-30T01:03:16.000-07:00
PiperOrigin-RevId: 668438132
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
@@ -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>
 }

Original file line number	Diff line number	Diff line change
`@@ -2,7 +2,7 @@`
`2`	`2`
`3`	`3`	`// CHECK-LABEL: func @pointwise_op`
`4`	`4`	`func.func @pointwise_op(%arg0: tensor<2x1x4xf32>) -> tensor<2x1x4xf32> {`
`5`		`- // CHECK: sdy.sharding_rule = #sdy.op_sharding_rule<([i, j, k], [i, j, k])->([i, j, k]) {i=2, j=1, k=4}>`
	`5`	`+ // CHECK: sdy.sharding_rule = #sdy.op_sharding_rule<([i, j, k], [i, j, k])->([i, j, k]) {i=3, j=1, k=4}>`
`6`	`6`	`%0 = stablehlo.add %arg0, %arg0: tensor<2x1x4xf32>`
`7`	`7`	`return %0 : tensor<2x1x4xf32>`
`8`	`8`	`}`