Upgrade Abseil to LTS 20260107.1

PiperOrigin-RevId: 866303896

Author: akuegel

MODULE.bazel

@@ -4,7 +4,7 @@ module(name = "xla")
 # Bazel module dependencies
 
 # go/keep-sorted start
-bazel_dep(name = "abseil-cpp", version = "20250814.1", repo_name = "com_google_absl")
+bazel_dep(name = "abseil-cpp", version = "20260107.1", repo_name = "com_google_absl")
 bazel_dep(name = "abseil-py", version = "2.1.0", repo_name = "absl_py")
 bazel_dep(name = "bazel_features", version = "1.36.0")
 bazel_dep(name = "bazel_skylib", version = "1.8.1")

third_party/absl/workspace.bzl

@@ -7,8 +7,8 @@ def repo():
 
     # Attention: tools parse and update these lines.
     # LINT.IfChange
-    ABSL_COMMIT = "d38452e1ee03523a208362186fd42248ff2609f6"  # LTS 20250814.1
-    ABSL_SHA256 = "d1abe9da2003e6cbbd7619b0ced3e52047422f4f4ac6c66a9bef5d2e99fea837"
+    ABSL_COMMIT = "255c84dadd029fd8ad25c5efb5933e47beaa00c7"  # LTS 20260107.1
+    ABSL_SHA256 = "87e91fb785a2d0233f4599317afd576b7736e6732d557bdcdfdc11990bd333ef"
     # LINT.ThenChange(//tensorflow/lite/tools/cmake/modules/abseil-cpp.cmake)
 
     tf_http_archive(

xla/debug_options_flags.cc

@@ -408,7 +408,7 @@ DebugOptions DefaultDebugOptionsIgnoringFlags() {
 
   opts.set_xla_gpu_nccl_terminate_on_error(false);
 
-  opts.set_xla_gpu_shard_autotuning(true);
+  opts.set_xla_gpu_shard_autotuning(false);
 
   opts.set_xla_syntax_sugar_async_ops(false);
 

xla/pjrt/gpu/tfrt/tfrt_gpu_client.cc

@@ -774,6 +774,7 @@ TfrtGpuClient::LoadInternal(
         std::move(hlo_module_proto),
         compile_options.executable_build_options.debug_options());
   }
+  LOG(ERROR) << "Returning Executable";
   return std::unique_ptr<PjRtLoadedExecutable>(std::move(executable));
 }
 
@@ -785,7 +786,7 @@ absl::StatusOr<std::unique_ptr<PjRtLoadedExecutable>> TfrtGpuClient::Load(
   CompileOptions compile_options = se_executable->compile_options();
 
   tsl::profiler::TraceMe traceme("TfrtGpuClient::Load");
-  VLOG(1) << "TfrtGpuClient::Load";
+  LOG(ERROR) << "TfrtGpuClient::Load";
 
   TF_ASSIGN_OR_RETURN(auto local_executable, se_executable->ConsumeExecutable(
                                                  xla_client_, compile_options));

xla/pjrt/pjrt_stream_executor_client.cc

@@ -2295,7 +2295,7 @@ PjRtStreamExecutorClient::CompileInternal(
     LayoutCanonicalizationCallback layout_canonicalization_callback,
     CompileOptions options, bool lookup_addressable_devices) {
   tsl::profiler::TraceMe traceme("PjRtStreamExecutorClient::CompileInternal");
-  VLOG(1) << "PjRtStreamExecutorClient::CompileInternal";
+  LOG(ERROR) << "PjRtStreamExecutorClient::CompileInternal";
   if (key_value_store().has_value() &&
       !options.executable_build_options.key_value_store()) {
     options.executable_build_options.set_key_value_store(*key_value_store());
@@ -2331,6 +2331,7 @@ PjRtStreamExecutorClient::CompileInternal(
     return Unimplemented("Multiple executables are not supported");
   }
 
+  LOG(ERROR) << "Returning executable";
   return BuildPjRtExecutable(xla_dump_hlo_unoptimized_snapshots
                                  ? std::make_optional(computation.proto())
                                  : std::nullopt,
@@ -2708,6 +2709,7 @@ PjRtStreamExecutorClient::LoadInternal(
         std::move(*unoptimized_hlo_module_proto),
         compile_options.executable_build_options.debug_options());
   }
+  LOG(ERROR) << "Returning loaded executable";
   return std::unique_ptr<PjRtLoadedExecutable>(std::move(executable));
 }
 
@@ -2719,7 +2721,7 @@ PjRtStreamExecutorClient::Load(std::unique_ptr<PjRtExecutable> executable,
   CompileOptions compile_options = se_executable->compile_options();
 
   tsl::profiler::TraceMe traceme("PjRtStreamExecutorClient::Load");
-  VLOG(1) << "PjRtStreamExecutorClient::Load";
+  LOG(ERROR) << "PjRtStreamExecutorClient::Load";
 
   TF_ASSIGN_OR_RETURN(auto local_executable, se_executable->ConsumeExecutable(
                                                  client(), compile_options));

xla/python/pjrt_ifrt/pjrt_executable.cc

@@ -758,10 +758,12 @@ absl::StatusOr<LoadedExecutableRef> PjRtLoadedExecutable::Create(
   TF_ASSIGN_OR_RETURN(const std::vector<xla::LayoutMode> output_layout_modes,
                       GetOutputLayoutModes(module));
 
+  LOG(ERROR) << "Calling Client CompileAndLoad";
   TF_ASSIGN_OR_RETURN(
       std::shared_ptr<xla::PjRtLoadedExecutable> pjrt_loaded_executable,
       client->pjrt_client()->CompileAndLoad(std::move(module),
                                             std::move(compile_options)));
+  LOG(ERROR) << "Client CompileAndLoad finished";
 
   TF_ASSIGN_OR_RETURN(
       executable_devices,

xla/service/compilation_stats.cc

@@ -132,9 +132,14 @@ void Stats::CompilationReport() {
   });
   LOG(INFO) << "Total runtime (ms) of HLO passes: " << total_duration;
   LOG(INFO) << "Pass name, num runs, time (ms)";
+  int64_t print_counter = 0;
   for (auto& pass_info : sorted_summary) {
     LOG(INFO) << pass_info.name << ", " << pass_info.num_runs << ", "
               << pass_info.duration_ms;
+    ++print_counter;
+    if (print_counter > 2) {
+      break;
+    }
   }
 }