diff --git a/build_tools/llvm_version.txt b/build_tools/llvm_version.txt index e3165a9d..ef7b51d4 100644 --- a/build_tools/llvm_version.txt +++ b/build_tools/llvm_version.txt @@ -1 +1 @@ -21830c913505b1fd2cf10e454253483180c7e10b +cca9f9b78fc657c280f7e4024a552af43a315bdb diff --git a/test/Conversion/stablehlo-to-emitc.mlir b/test/Conversion/stablehlo-to-emitc.mlir index 6215df44..c91df699 100644 --- a/test/Conversion/stablehlo-to-emitc.mlir +++ b/test/Conversion/stablehlo-to-emitc.mlir @@ -333,8 +333,8 @@ func.func @stablehlo_conv(%arg0: tensor<3x2x4x3xf32>, %arg1 : tensor<2x2x3x4xf32 >, feature_group_count = 1 : i64, padding = dense<[[0, 1], [0, 1]]> : tensor<2x2xi64>, - rhs_dilation = dense<[1, 2]> : tensor<2xi64>, - window_strides = dense<[2, 1]> : tensor<2xi64> + rhs_dilation = array, + window_strides = array } : (tensor<2x2x3x4xf32>, tensor<3x2x4x3xf32>) -> tensor<2x1x2x3xf32> return %out : tensor<2x1x2x3xf32> } @@ -369,14 +369,14 @@ func.func @stablehlo_reduce(%arg0 : tensor<2x1000xf32>, %arg1 : tensor, %ar ^bb0(%arg4: tensor, %arg5: tensor): %1 = stablehlo.add %arg4, %arg5 : tensor "stablehlo.return"(%1) : (tensor) -> () - }) {dimensions = dense<1> : tensor<1xi64>} : (tensor<2x1000xf32>, tensor) -> tensor<2xf32> + }) {dimensions = array} : (tensor<2x1000xf32>, tensor) -> tensor<2xf32> // CHECK: emitc.call_opaque "emitc::stablehlo::reduce"(%arg2, %arg3) {args = [0 : index, 1 : index, dense<1> : tensor<1xi64>, @stablehlo_reduce_lambda_1], template_args = [tensor<2xi32>, 1]} : (tensor<2x1000xi32>, tensor) -> tensor<2xi32> %1 = "stablehlo.reduce"(%arg2, %arg3) ({ ^bb0(%arg4: tensor, %arg5: tensor): %2 = stablehlo.maximum %arg4, %arg5 : tensor "stablehlo.return"(%2) : (tensor) -> () - }) {dimensions = dense<1> : tensor<1xi64>} : (tensor<2x1000xi32>, tensor) -> tensor<2xi32> + }) {dimensions = array} : (tensor<2x1000xi32>, tensor) -> tensor<2xi32> // CHECK: emitc.call_opaque "emitc::stablehlo::reduce"(%arg0, %arg2, %arg1, %arg3) {args = [0 : index, 1 : index, 2 : index, 3 : index, dense<1> : tensor<1xi64>, @stablehlo_reduce_lambda_2], template_args = [tensor<2xf32>, tensor<2xi32>, 1]} : (tensor<2x1000xf32>, tensor<2x1000xi32>, tensor, tensor) -> (tensor<2xf32>, tensor<2xi32>) %2:2 = stablehlo.reduce(%arg0 init: %arg1), (%arg2 init: %arg3) across dimensions = [1] : (tensor<2x1000xf32>, tensor<2x1000xi32>, tensor, tensor) -> (tensor<2xf32>, tensor<2xi32>) @@ -397,7 +397,7 @@ func.func @stablehlo_reduce_window(%arg0 : tensor<2x114x114x64xf32>, %arg1 : ten ^bb0(%arg2: tensor, %arg3: tensor): // no predecessors %516 = stablehlo.maximum %arg2, %arg3 : tensor "stablehlo.return"(%516) : (tensor) -> () - }) {window_dimensions = dense<[1, 3, 3, 1]> : tensor<4xi64>, window_strides = dense<[1, 2, 2, 1]> : tensor<4xi64>} : (tensor<2x114x114x64xf32>, tensor) -> tensor<2x56x56x64xf32> + }) {window_dimensions = array, window_strides = array} : (tensor<2x114x114x64xf32>, tensor) -> tensor<2x56x56x64xf32> return %0 : tensor<2x56x56x64xf32> } diff --git a/test/MobileNetV2_FakeWeights_stablehlo.mlir b/test/MobileNetV2_FakeWeights_stablehlo.mlir index 32bdabc6..1c3d4d2d 100644 --- a/test/MobileNetV2_FakeWeights_stablehlo.mlir +++ b/test/MobileNetV2_FakeWeights_stablehlo.mlir @@ -204,7 +204,7 @@ module attributes {tf.versions = {bad_consumers = [], min_consumer = 12 : i32, p ^bb0(%arg1: tensor, %arg2: tensor): // no predecessors %210 = stablehlo.add %arg1, %arg2 : tensor "stablehlo.return"(%210) : (tensor) -> () - }) {dimensions = dense<[1, 2]> : tensor<2xi64>} : (tensor<1x7x7x1280xf32>, tensor) -> tensor<1x1280xf32> + }) {dimensions = array} : (tensor<1x7x7x1280xf32>, tensor) -> tensor<1x1280xf32> %200 = stablehlo.divide %199, %1 : tensor<1x1280xf32> %201 = "stablehlo.dot"(%200, %9) : (tensor<1x1280xf32>, tensor<1280x1000xf32>) -> tensor<1x1000xf32> %202 = stablehlo.add %201, %0 : tensor<1x1000xf32> @@ -212,7 +212,7 @@ module attributes {tf.versions = {bad_consumers = [], min_consumer = 12 : i32, p ^bb0(%arg1: tensor, %arg2: tensor): // no predecessors %210 = stablehlo.maximum %arg1, %arg2 : tensor "stablehlo.return"(%210) : (tensor) -> () - }) {dimensions = dense<1> : tensor<1xi64>} : (tensor<1x1000xf32>, tensor) -> tensor<1xf32> + }) {dimensions = array} : (tensor<1x1000xf32>, tensor) -> tensor<1xf32> %204 = "stablehlo.broadcast_in_dim"(%203) {broadcast_dimensions = array} : (tensor<1xf32>) -> tensor<1x1000xf32> %205 = stablehlo.subtract %202, %204 : tensor<1x1000xf32> %206 = "stablehlo.exponential"(%205) : (tensor<1x1000xf32>) -> tensor<1x1000xf32> @@ -220,7 +220,7 @@ module attributes {tf.versions = {bad_consumers = [], min_consumer = 12 : i32, p ^bb0(%arg1: tensor, %arg2: tensor): // no predecessors %210 = stablehlo.add %arg1, %arg2 : tensor "stablehlo.return"(%210) : (tensor) -> () - }) {dimensions = dense<1> : tensor<1xi64>} : (tensor<1x1000xf32>, tensor) -> tensor<1xf32> + }) {dimensions = array} : (tensor<1x1000xf32>, tensor) -> tensor<1xf32> %208 = "stablehlo.broadcast_in_dim"(%207) {broadcast_dimensions = array} : (tensor<1xf32>) -> tensor<1x1000xf32> %209 = stablehlo.divide %206, %208 : tensor<1x1000xf32> return %209 : tensor<1x1000xf32> diff --git a/third_party/stablehlo b/third_party/stablehlo index 85d9ad56..e191eb4c 160000 --- a/third_party/stablehlo +++ b/third_party/stablehlo @@ -1 +1 @@ -Subproject commit 85d9ad563f0d1672c8532d7abc886778223bdf45 +Subproject commit e191eb4c3c3f3144503a8a117d760de5ddcc7e89