Enable multi-host support for trace viewer.
PiperOrigin-RevId: 821973423
diff --git a/third_party/xla/xla/tsl/profiler/utils/trace_utils.h b/third_party/xla/xla/tsl/profiler/utils/trace_utils.h
index f4d82a0..8f0db91 100644
--- a/third_party/xla/xla/tsl/profiler/utils/trace_utils.h
+++ b/third_party/xla/xla/tsl/profiler/utils/trace_utils.h
@@ -16,6 +16,7 @@
#ifndef XLA_TSL_PROFILER_UTILS_TRACE_UTILS_H_
#define XLA_TSL_PROFILER_UTILS_TRACE_UTILS_H_
+#include <cstdint>
#include <optional>
#include "absl/strings/numbers.h"
@@ -30,6 +31,9 @@
// Support up to 500 accelerator devices.
constexpr uint32 kFirstDeviceId = 1;
constexpr uint32 kLastDeviceId = 500;
+
+// Max. devices per host. Power of 10 for ease of debugging.
+static constexpr uint32_t kMaxDevicesPerHost = 1000;
// Support Upto 200 custom planes as fake devices (i.e., planes with a
// "/custom:" prefix). See `<project_name>::kCustomPlanePrefix` for more
// information
