Integrate LLVM at llvm/llvm-project@3a82a1c3f6bd (#1932)

Also accommodating changes related to 

-
llvm/llvm-project@5fcf907
-
https://github.com/llvm/llvm-project/pull/78376/files#diff-04e8290220df887ebaa6c4bf8ba9d121af3df7536732e447029b6f342b5aaa62

WORKSPACE.bazel

@@ -28,9 +28,9 @@ http_archive(
     ],
 )
 
-LLVM_COMMIT = "baba0a4cb43181a78881fce683e3a5016daa8ce6"
+LLVM_COMMIT = "3a82a1c3f6bdc9259cc4641f66fc76d1e171e382"
 
-LLVM_SHA256 = "a81c8c08b7fc11a9668b2ed3e37a3e98ad8f9e4e4f6ba2c8b0b36e105a775d4e"
+LLVM_SHA256 = "c525cdb14bb239695852d696bcd13a6d47e579be18386ba2048515fe7f059153"
 
 http_archive(
     name = "llvm-raw",

build_tools/llvm_version.txt

@@ -1 +1 @@
-baba0a4cb43181a78881fce683e3a5016daa8ce6
+3a82a1c3f6bdc9259cc4641f66fc76d1e171e382

stablehlo/conversions/linalg/tests/reduce.mlir

@@ -29,7 +29,7 @@ func.func @reduce_add(%arg0: tensor<5x4xi32>, %arg1: tensor<i32>) -> tensor<5xi3
 // CHECK-PRIMITIVE-DAG: %[[INIT:.*]] = tensor.extract %{{.*}} : tensor<i32>
 // CHECK-PRIMITIVE-DAG: %[[INIT_TENSOR:.*]] = tensor.empty()
 // CHECK-PRIMITIVE-DAG: %[[FILL_TENSOR:.*]] = linalg.fill ins(%[[INIT]]{{.*}}outs(%[[INIT_TENSOR]]
-// CHECK-PRIMITIVE: linalg.reduce { arith.addi }
+// CHECK-PRIMITIVE: linalg.reduce { arith.addi {overflowFlags = #arith.overflow<none>} }
 // CHECK-PRIMITIVE-SAME: ins(%{{.*}}tensor<5x4xi32>)
 // CHECK-PRIMITIVE-SAME: outs(%[[FILL_TENSOR]] : tensor<5xi32>)
 // CHECK-PRIMITIVE-SAME: dimensions = [1]  {someattr}

stablehlo/transforms/StablehloRefineShapes.cpp

@@ -967,16 +967,16 @@ struct RefineWhileOpPattern : public OpRewritePattern<WhileOp> {
     // better error reporting for this case.
     // This serves the current use cases well, so the implementation of more
     // sophisticated refinement algorithm is left for future work.
-    rewriter.startRootUpdate(op);
+    rewriter.startOpModification(op);
     auto condStatus = refineValues(rewriter, op, op.getCond().getArguments(),
                                    op.getOperandTypes());
     auto bodyStatus = refineValues(rewriter, op, op.getBody().getArguments(),
                                    op.getOperandTypes());
     if (succeeded(condStatus) || succeeded(bodyStatus)) {
-      rewriter.finalizeRootUpdate(op);
+      rewriter.finalizeOpModification(op);
       return success();
     } else {
-      rewriter.cancelRootUpdate(op);
+      rewriter.cancelOpModification(op);
       return failure();
     }
   }
@@ -1055,7 +1055,7 @@ struct UpdateRegionTypePattern : public OpRewritePattern<ReturnOp> {
     if (!needsUpdate)
       return rewriter.notifyMatchFailure(op, "doesn't need update");
 
-    rewriter.updateRootInPlace(op->getParentOp(), [&]() { return; });
+    rewriter.modifyOpInPlace(op->getParentOp(), [&]() { return; });
     return success();
   }
 };