From d8baecaf9a1b958fbeb84beef6d0b6b7820b1eb8 Mon Sep 17 00:00:00 2001 From: justinlu Date: Mon, 24 Aug 2026 10:59:26 -0700 Subject: [PATCH] Fix compilation errors and test failures across tpu_sync targets PiperOrigin-RevId: 969954875 --- tpu_sync/frameworks/torch/BUILD | 1 + tpu_sync/telemetry/prometheus_exporter.cc | 3 +++ 2 files changed, 4 insertions(+) diff --git a/tpu_sync/frameworks/torch/BUILD b/tpu_sync/frameworks/torch/BUILD index 5f21dab4..5f0e7dfd 100644 --- a/tpu_sync/frameworks/torch/BUILD +++ b/tpu_sync/frameworks/torch/BUILD @@ -429,6 +429,7 @@ cc_library( tags = ["nobuilder"], visibility = ["//visibility:public"], deps = [ + ":torch_tpu_device_buffer_headers", ":torch_tpu_tensor_to_buffer_headers", ":torch_utils_mock", "//tpu_sync/weight_sync:weight_synchronizer_base", diff --git a/tpu_sync/telemetry/prometheus_exporter.cc b/tpu_sync/telemetry/prometheus_exporter.cc index 66882975..a62e9170 100644 --- a/tpu_sync/telemetry/prometheus_exporter.cc +++ b/tpu_sync/telemetry/prometheus_exporter.cc @@ -75,6 +75,7 @@ void PrometheusExporter::RegisterKnownFamilies() { .Name(prometheus_name) .Help(std::string(meta.description)) .Register(*registry_); + family->Add({}); counter_families_.emplace(meta.name, family); break; } @@ -83,6 +84,7 @@ void PrometheusExporter::RegisterKnownFamilies() { .Name(prometheus_name) .Help(std::string(meta.description)) .Register(*registry_); + family->Add({}); gauge_families_.emplace(meta.name, family); break; } @@ -91,6 +93,7 @@ void PrometheusExporter::RegisterKnownFamilies() { .Name(prometheus_name) .Help(std::string(meta.description)) .Register(*registry_); + family->Add({}, default_buckets_); histogram_families_.emplace(meta.name, family); break; }