Integrate LLVM at llvm/llvm-project@17ba4f4053e3

Updates LLVM usage to match
[17ba4f4053e3](llvm/llvm-project@17ba4f4053e3)

PiperOrigin-RevId: 658383204

Author: gribozavr

tensorflow/compiler/mlir/lite/transforms/lower_static_tensor_list.cc

@@ -81,25 +81,6 @@ namespace mlir {
 
 namespace {
 
-// TODO(b/355062942): This a temporary solution to unblock LLVM intergration.
-// https://github.com/llvm/llvm-project/commit/bbd4af5da2b741672a8e6f625eb12ea5c2d6220f
-// changed the behavior of `applySignatureConversion`. Before, an op adaptor
-// would have the new block arguments directly as operands. Now, there is an
-// `UnrealizedConversionCastOp` inserts from the new type to the old type.
-// The new behaviour is correct, but passes in this file depended on the old
-// bahavior and worked by coincidence.
-llvm::SmallVector<Value, 4> GetOperandsAndSkipUnrealizedConversionCasts(
-    ValueRange operands) {
-  llvm::SmallVector<Value, 4> result;
-  for (Value operand : operands) {
-    if (auto cast = operand.getDefiningOp<UnrealizedConversionCastOp>()) {
-      operand = cast.getInputs().front();
-    }
-    result.push_back(operand);
-  }
-  return result;
-}
-
 /// Lower TensorList ops in functions for subsequent legalization.
 struct LowerStaticTensorListPass
     : public impl::LowerStaticTensorListPassBase<LowerStaticTensorListPass> {
@@ -371,9 +352,7 @@ struct ConvertTensorListSetItem
       ConversionPatternRewriter &rewriter) const {
     Location loc = op.getLoc();
 
-    auto operands =
-        GetOperandsAndSkipUnrealizedConversionCasts(adaptor.getOperands());
-
+    auto operands = adaptor.getOperands();
     Value input = operands[0];
     Value index = operands[1];
     Value item = operands[2];
@@ -433,8 +412,7 @@ struct ConvertTensorListSetItem
       ConversionPatternRewriter &rewriter) const {
     Location loc = op.getLoc();
 
-    auto operands =
-        GetOperandsAndSkipUnrealizedConversionCasts(adaptor.getOperands());
+    auto operands = adaptor.getOperands();
     Value input = operands[0];
     Value index = operands[1];
     Value item = operands[2];
@@ -721,8 +699,7 @@ struct ConvertTensorListPushBack
   LogicalResult matchAndRewrite(
       TF::TensorListPushBackOp op, OpAdaptor adaptor,
       ConversionPatternRewriter &rewriter) const override {
-    auto operands =
-        GetOperandsAndSkipUnrealizedConversionCasts(adaptor.getOperands());
+    auto operands = adaptor.getOperands();
     Value input_handle = operands[0];
     Value item = operands[1];
 
@@ -764,8 +741,7 @@ struct ConvertTensorListResize
   LogicalResult matchAndRewrite(
       TF::TensorListResizeOp op, OpAdaptor adaptor,
       ConversionPatternRewriter &rewriter) const override {
-    auto operands =
-        GetOperandsAndSkipUnrealizedConversionCasts(adaptor.getOperands());
+    auto operands = adaptor.getOperands();
     Value input_handle = operands[0];
     Value size = operands[1];
 
@@ -929,9 +905,7 @@ struct ConvertTensorListGetItem
   LogicalResult matchAndRewrite(
       TF::TensorListGetItemOp op, OpAdaptor adaptor,
       ConversionPatternRewriter &rewriter) const override {
-    auto operands =
-        GetOperandsAndSkipUnrealizedConversionCasts(adaptor.getOperands());
-
+    auto operands = adaptor.getOperands();
     Value input = operands[0];
     Value index = operands[1];
     rewriter.replaceOpWithNewOp<TF::GatherOp>(op, op.getType(), input, index,
@@ -948,8 +922,7 @@ struct ConvertTensorListLength
       TF::TensorListLengthOp op, OpAdaptor adaptor,
       ConversionPatternRewriter &rewriter) const override {
     Location loc = op.getLoc();
-    Value input_handle =
-        GetOperandsAndSkipUnrealizedConversionCasts(adaptor.getOperands())[0];
+    Value input_handle = adaptor.getOperands()[0];
 
     BoolAttr true_attr = rewriter.getBoolAttr(true);
     auto shape = rewriter.create<TF::ShapeOp>(loc, input_handle,
@@ -970,8 +943,7 @@ struct ConvertTensorListStack
       ConversionPatternRewriter &rewriter) const override {
     Location loc = op.getLoc();
 
-    auto operands =
-        GetOperandsAndSkipUnrealizedConversionCasts(adaptor.getOperands());
+    auto operands = adaptor.getOperands();
     Value input = operands[0];
     Value element_shape = operands[1];
 
@@ -1021,8 +993,7 @@ struct ConvertTensorListConcatV2
       ConversionPatternRewriter &rewriter) const override {
     Location loc = op.getLoc();
 
-    auto operands =
-        GetOperandsAndSkipUnrealizedConversionCasts(adaptor.getOperands());
+    auto operands = adaptor.getOperands();
     Value input = operands[0];
     Value element_shape = operands[1];
 
@@ -1084,8 +1055,7 @@ struct ConvertIdentity : public OpConversionPattern<TF::IdentityOp> {
   LogicalResult matchAndRewrite(
       TF::IdentityOp op, OpAdaptor adaptor,
       ConversionPatternRewriter &rewriter) const override {
-    Value input =
-        GetOperandsAndSkipUnrealizedConversionCasts(adaptor.getOperands())[0];
+    Value input = adaptor.getOperands()[0];
     rewriter.replaceOpWithNewOp<TF::IdentityOp>(op, input.getType(), input,
                                                 op->getAttrs());
     return success();
@@ -1098,9 +1068,7 @@ struct ConvertReturn : public OpConversionPattern<func::ReturnOp> {
   LogicalResult matchAndRewrite(
       func::ReturnOp op, OpAdaptor adaptor,
       ConversionPatternRewriter &rewriter) const override {
-    auto operands =
-        GetOperandsAndSkipUnrealizedConversionCasts(adaptor.getOperands());
-
+    auto operands = adaptor.getOperands();
     rewriter.replaceOpWithNewOp<func::ReturnOp>(op, ValueRange{}, operands,
                                                 op->getAttrs());
     return success();
@@ -1113,8 +1081,7 @@ struct ConvertYield : public OpConversionPattern<TF::YieldOp> {
   LogicalResult matchAndRewrite(
       TF::YieldOp op, OpAdaptor adaptor,
       ConversionPatternRewriter &rewriter) const override {
-    auto operands =
-        GetOperandsAndSkipUnrealizedConversionCasts(adaptor.getOperands());
+    auto operands = adaptor.getOperands();
     rewriter.replaceOpWithNewOp<TF::YieldOp>(op, operands);
     return success();
   }

third_party/llvm/workspace.bzl

@@ -4,8 +4,8 @@ load("//third_party:repo.bzl", "tf_http_archive")
 
 def repo(name):
     """Imports LLVM."""
-    LLVM_COMMIT = "42d641ef5cc4bd82f98ef9959a593ca6db66d75d"
-    LLVM_SHA256 = "ec368e9c3b1e1c5eb646c21da65bb54a53060b417e61f2451f3917b35d743abd"
+    LLVM_COMMIT = "17ba4f4053e303be3e5408d34eaf687a49cefb06"
+    LLVM_SHA256 = "64c334c15f058ca090fe8eb1e2cd99fdcbaaaf6e57202760f32574d3a9d24d78"
 
     tf_http_archive(
         name = name,

third_party/shardy/temporary.patch

@@ -1,15 +1,15 @@
 diff --git a/third_party/llvm/workspace.bzl b/third_party/llvm/workspace.bzl
-index 9345d8d..6429d9b 100644
+index 6429d9b..e0f0b45 100644
 --- a/third_party/llvm/workspace.bzl
 +++ b/third_party/llvm/workspace.bzl
 @@ -4,8 +4,8 @@ load("//third_party:repo.bzl", "tf_http_archive")
 
  def repo(name):
      """Imports LLVM."""
--    LLVM_COMMIT = "d92a484e6f5c9063d82ca79405bb3557d88ad575"
--    LLVM_SHA256 = "0e6cce920f7344248ed747443fc16c316faf398e33f6a7f9f11f41ede861f824"
-+    LLVM_COMMIT = "42d641ef5cc4bd82f98ef9959a593ca6db66d75d"
-+    LLVM_SHA256 = "ec368e9c3b1e1c5eb646c21da65bb54a53060b417e61f2451f3917b35d743abd"
+-    LLVM_COMMIT = "42d641ef5cc4bd82f98ef9959a593ca6db66d75d"
+-    LLVM_SHA256 = "ec368e9c3b1e1c5eb646c21da65bb54a53060b417e61f2451f3917b35d743abd"
++    LLVM_COMMIT = "17ba4f4053e303be3e5408d34eaf687a49cefb06"
++    LLVM_SHA256 = "64c334c15f058ca090fe8eb1e2cd99fdcbaaaf6e57202760f32574d3a9d24d78"
 
      tf_http_archive(
          name = name,

third_party/shardy/workspace.bzl

@@ -3,8 +3,8 @@
 load("//third_party:repo.bzl", "tf_http_archive", "tf_mirror_urls")
 
 def repo():
-    SHARDY_COMMIT = "df54e37427b0007e6527b62616ed1f66a68dda4a"
-    SHARDY_SHA256 = "2ebf03fd73c4578e721c539ad05b33d5fbfae6838abbb58b944e12f1eafbd9b2"
+    SHARDY_COMMIT = "a0d337eecb4957da862b235a2829efd9513a129c"
+    SHARDY_SHA256 = "ffbc55d51995da6fd149e7bc0e41bee4faba02dfa2984f2af54940c46578c0c7"
 
     tf_http_archive(
         name = "shardy",

third_party/xla/third_party/shardy/temporary.patch

@@ -1,15 +1,15 @@
 diff --git a/third_party/llvm/workspace.bzl b/third_party/llvm/workspace.bzl
-index 9345d8d..6429d9b 100644
+index 6429d9b..e0f0b45 100644
 --- a/third_party/llvm/workspace.bzl
 +++ b/third_party/llvm/workspace.bzl
 @@ -4,8 +4,8 @@ load("//third_party:repo.bzl", "tf_http_archive")
 
  def repo(name):
      """Imports LLVM."""
--    LLVM_COMMIT = "d92a484e6f5c9063d82ca79405bb3557d88ad575"
--    LLVM_SHA256 = "0e6cce920f7344248ed747443fc16c316faf398e33f6a7f9f11f41ede861f824"
-+    LLVM_COMMIT = "42d641ef5cc4bd82f98ef9959a593ca6db66d75d"
-+    LLVM_SHA256 = "ec368e9c3b1e1c5eb646c21da65bb54a53060b417e61f2451f3917b35d743abd"
+-    LLVM_COMMIT = "42d641ef5cc4bd82f98ef9959a593ca6db66d75d"
+-    LLVM_SHA256 = "ec368e9c3b1e1c5eb646c21da65bb54a53060b417e61f2451f3917b35d743abd"
++    LLVM_COMMIT = "17ba4f4053e303be3e5408d34eaf687a49cefb06"
++    LLVM_SHA256 = "64c334c15f058ca090fe8eb1e2cd99fdcbaaaf6e57202760f32574d3a9d24d78"
 
      tf_http_archive(
          name = name,

third_party/xla/third_party/shardy/workspace.bzl

@@ -3,8 +3,8 @@
 load("//third_party:repo.bzl", "tf_http_archive", "tf_mirror_urls")
 
 def repo():
-    SHARDY_COMMIT = "df54e37427b0007e6527b62616ed1f66a68dda4a"
-    SHARDY_SHA256 = "2ebf03fd73c4578e721c539ad05b33d5fbfae6838abbb58b944e12f1eafbd9b2"
+    SHARDY_COMMIT = "a0d337eecb4957da862b235a2829efd9513a129c"
+    SHARDY_SHA256 = "ffbc55d51995da6fd149e7bc0e41bee4faba02dfa2984f2af54940c46578c0c7"
 
     tf_http_archive(
         name = "shardy",