From 52b41bd227b6f280942f04d92076116e62b168e5 Mon Sep 17 00:00:00 2001 From: "A. Unique TensorFlower" Date: Thu, 8 Aug 2024 23:21:03 -0700 Subject: [PATCH] Automated Code Change PiperOrigin-RevId: 661136198 --- tensorflow/dtensor/mlir/utils/update_tpu_metadata.cc | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tensorflow/dtensor/mlir/utils/update_tpu_metadata.cc b/tensorflow/dtensor/mlir/utils/update_tpu_metadata.cc index 802d46fd27ecde..2e24e5d1f9db4c 100644 --- a/tensorflow/dtensor/mlir/utils/update_tpu_metadata.cc +++ b/tensorflow/dtensor/mlir/utils/update_tpu_metadata.cc @@ -165,7 +165,7 @@ Status UpdateMetadataProtoXlaSpmd(const Mesh& mesh_config, mesh_name = ""; } const std::vector& tpu_core_ids = Mesh::tpu_core_ids()[mesh_name]; - VLOG(1) << "tpu_core_ids: " << str_util::Join(tpu_core_ids, ", "); + VLOG(1) << "tpu_core_ids: " << absl::StrJoin(tpu_core_ids, ", "); xla::DeviceAssignmentProto device_assignment; device_assignment.set_replica_count(1); @@ -223,7 +223,7 @@ Status UpdateMetadataProtoDtensorSpmd(const Mesh& mesh_config, mesh_name = ""; } const std::vector& tpu_core_ids = Mesh::tpu_core_ids()[mesh_name]; - VLOG(1) << "tpu_core_ids: " << str_util::Join(tpu_core_ids, ", "); + VLOG(1) << "tpu_core_ids: " << absl::StrJoin(tpu_core_ids, ", "); xla::DeviceAssignmentProto device_assignment; device_assignment.set_replica_count(num_replicas);