From e3b1feae1588243cfd8bf968d26f7122b7467be0 Mon Sep 17 00:00:00 2001
From: Praneeth Mandala <praneman@google.com>
Date: Mon, 3 Nov 2025 22:51:35 -0800
Subject: [PATCH] Trying A/B tests for previous bugs.

PiperOrigin-RevId: 827801139
---
 xprof/convert/xplane_to_tools_data.cc | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/xprof/convert/xplane_to_tools_data.cc b/xprof/convert/xplane_to_tools_data.cc
index f8bc5e71..63b62555 100644
--- a/xprof/convert/xplane_to_tools_data.cc
+++ b/xprof/convert/xplane_to_tools_data.cc
@@ -564,6 +564,11 @@ absl::StatusOr<std::string> ConvertMultiXSpacesToToolData(
   } else if (tool_name == "megascale_stats") {
     return ConvertDcnCollectiveStatsToToolData(session_snapshot, options);
   } else if (tool_name == "tool_names") {
+    TF_ASSIGN_OR_RETURN(bool hlo_proto_status,
+                        ConvertMultiXSpaceToHloProto(session_snapshot));
+    if (!hlo_proto_status) {
+      return absl::NotFoundError("No HLO proto found in XSpace.");
+    }
     return GetAvailableToolNames(session_snapshot);
   } else if (tool_name == "_xplane.pb") {  // internal test only.
     return PreprocessXSpace(session_snapshot);