diff --git a/xprof/convert/xplane_to_tools_data.cc b/xprof/convert/xplane_to_tools_data.cc index f8bc5e712..63b62555b 100644 --- a/xprof/convert/xplane_to_tools_data.cc +++ b/xprof/convert/xplane_to_tools_data.cc @@ -564,6 +564,11 @@ absl::StatusOr 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);