Skip to content

Commit 048f9d5

Browse files
hhbtensorflower-gardener
authored andcommitted
Add logs for coordination service initialization.
PiperOrigin-RevId: 614839485
1 parent db79be2 commit 048f9d5

File tree

2 files changed

+3
-1
lines changed

2 files changed

+3
-1
lines changed

third_party/xla/third_party/tsl/tsl/distributed_runtime/coordination/BUILD

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -81,7 +81,7 @@ tsl_gpu_library(
8181
"//tsl/util:device_name_utils",
8282
"@com_google_absl//absl/container:flat_hash_map",
8383
"@com_google_absl//absl/container:flat_hash_set",
84-
"@com_google_absl//absl/memory",
84+
"@com_google_absl//absl/log",
8585
"@com_google_absl//absl/strings",
8686
"@com_google_absl//absl/synchronization",
8787
"@com_google_absl//absl/time",

third_party/xla/third_party/tsl/tsl/distributed_runtime/coordination/coordination_service.cc

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -26,6 +26,7 @@ limitations under the License.
2626

2727
#include "absl/container/flat_hash_map.h"
2828
#include "absl/container/flat_hash_set.h"
29+
#include "absl/log/log.h"
2930
#include "absl/strings/str_cat.h"
3031
#include "absl/strings/str_join.h"
3132
#include "absl/strings/string_view.h"
@@ -361,6 +362,7 @@ CoordinationServiceStandaloneImpl::CoordinationServiceStandaloneImpl(
361362
absl::Milliseconds(config.shutdown_barrier_timeout_in_ms())),
362363
allow_new_incarnation_to_reconnect_(
363364
config.allow_new_incarnation_to_reconnect()) {
365+
LOG(INFO) << "Initializing CoordinationService";
364366
recoverable_jobs_ = absl::flat_hash_set<std::string>(
365367
config.recoverable_jobs().cbegin(), config.recoverable_jobs().cend());
366368
for (const auto& job : config.coordinated_job_list()) {

0 commit comments

Comments
 (0)