Switch to using bytes field for CoreDetails instead of string.

PiperOrigin-RevId: 712977029
GesuBackups · Jan 7, 2025 · ba2ddb9 · ba2ddb9
1 parent 4de05bd
commit ba2ddb9
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/tensorflow/core/profiler/convert/xplane_to_op_stats.cc b/tensorflow/core/profiler/convert/xplane_to_op_stats.cc
@@ -370,10 +370,12 @@ OpStats ConvertXSpaceToOpStats(const XSpace& space,
       auto stat = visitor.GetStat(StatType::kCoreDetails);
       if (stat.has_value()) {
         CoreDetails core_details;
-        // TODO: Switch to StrOrRefValue once protobuf version is updated.
-        core_details.ParseFromString(stat->ToString());
-        core_details.set_hostname(hostname);
-        core_id_to_details[device_plane->id()] = core_details;
+        absl::string_view core_details_bytes = stat->BytesValue();
+        if (core_details.ParseFromArray(core_details_bytes.data(),
+                                        core_details_bytes.size())) {
+          core_details.set_hostname(hostname);
+          core_id_to_details[device_plane->id()] = core_details;
+        }
       }
     }
   }