Integrate LLVM at llvm/llvm-project@a26cbd03fe0f

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

PiperOrigin-RevId: 499930513

Author: durin42

third_party/llvm/workspace.bzl

@@ -4,8 +4,8 @@ load("//third_party:repo.bzl", "tf_http_archive")
 
 def repo(name):
     """Imports LLVM."""
-    LLVM_COMMIT = "26fec4e845a833214d2b0036ea90f1b1c2418d50"
-    LLVM_SHA256 = "07b96c42044754e0cff3c976262528db8cc8a9264b89486716feb46c9a7a0cf8"
+    LLVM_COMMIT = "a26cbd03fe0f6f996d27f793e2107393d1888e40"
+    LLVM_SHA256 = "e05af776d2c3a5973020f3787d692c330db8da0da117ef097644881d4bea28c3"
 
     tf_http_archive(
         name = name,

xla/mlir_hlo/gml_st/transforms/peeling/peeling.cc

@@ -120,21 +120,17 @@ LogicalResult peelLoop(RewriterBase &b, LoopTy loopOp, int64_t idx,
   return success();
 }
 
-template <typename OpTy, bool IsMin>
+template <typename OpTy>
 void rewriteAffineOpAfterPeeling(RewriterBase &rewriter, Operation *mainLoop,
                                  Operation *remainderLoop, Value mainIv,
                                  Value remainderIv, Value ub, Value step) {
   mainLoop->walk([&](OpTy affineOp) {
-    AffineMap map = affineOp.getAffineMap();
-    (void)scf::rewritePeeledMinMaxOp(rewriter, affineOp, map,
-                                     affineOp.getOperands(), IsMin, mainIv, ub,
-                                     step, /*insideLoop=*/true);
+    (void)scf::rewritePeeledMinMaxOp(rewriter, affineOp, mainIv, ub, step,
+                                     /*insideLoop=*/true);
   });
   remainderLoop->walk([&](OpTy affineOp) {
-    AffineMap map = affineOp.getAffineMap();
-    (void)scf::rewritePeeledMinMaxOp(rewriter, affineOp, map,
-                                     affineOp.getOperands(), IsMin, remainderIv,
-                                     ub, step, /*insideLoop=*/false);
+    (void)scf::rewritePeeledMinMaxOp(rewriter, affineOp, remainderIv, ub, step,
+                                     /*insideLoop=*/false);
   });
 }
 
@@ -155,10 +151,10 @@ FailureOr<LoopTy> peelAndCanonicalizeGmlStLoopImpl(RewriterBase &rewriter,
   Value mainIv = loopOp.getInductionVars()[idx], step = loopOp.getStep()[idx],
         remainderIv = remainderLoop.getInductionVars()[idx];
 
-  rewriteAffineOpAfterPeeling<AffineMinOp, /*IsMin=*/true>(
-      rewriter, loopOp, remainderLoop, mainIv, remainderIv, ub, step);
-  rewriteAffineOpAfterPeeling<AffineMaxOp, /*IsMin=*/false>(
-      rewriter, loopOp, remainderLoop, mainIv, remainderIv, ub, step);
+  rewriteAffineOpAfterPeeling<AffineMinOp>(rewriter, loopOp, remainderLoop,
+                                           mainIv, remainderIv, ub, step);
+  rewriteAffineOpAfterPeeling<AffineMaxOp>(rewriter, loopOp, remainderLoop,
+                                           mainIv, remainderIv, ub, step);
 
   return remainderLoop;
 }

xla/mlir_hlo/tests/Dialect/gml_st/canonicalize.mlir

@@ -1,4 +1,4 @@
-// RUN: mlir-hlo-opt %s -canonicalize -split-input-file | FileCheck %s
+// RUN: mlir-hlo-opt %s -canonicalize="test-convergence" -split-input-file | FileCheck %s
 
 #map = affine_map<(d0)[s0, s1] -> (d0 * s1 + s0)>