Project import generated by Copybara

PiperOrigin-RevId: 829261364

Author: Profiler Team

xprof/convert/multi_xplanes_to_op_stats.cc

@@ -62,8 +62,8 @@ absl::Status ConvertMultiXSpacesToCombinedOpStats(
   }
 
   // Do not limit the maximum number of steps during the merge of OpStats.
-  StepIntersection step_intersection =
-      ComputeStepIntersectionToMergeOpStats(all_op_stats_info, kuint32max);
+  StepIntersection step_intersection = ComputeStepIntersectionToMergeOpStats(
+      all_op_stats_info, std::numeric_limits<uint32_t>::max());
   CombineAllOpStats(all_op_stats_info, step_intersection, combined_op_stats);
 
   return absl::OkStatus();

xprof/convert/op_metrics_db_combiner.cc

@@ -99,7 +99,7 @@ void CombineMemoryAccessedBreakdown(
     const tsl::protobuf::RepeatedPtrField<OpMetrics_MemoryAccessed>& src,
     tsl::protobuf::RepeatedPtrField<OpMetrics_MemoryAccessed>* dst) {
   if (src.empty()) return;
-  absl::flat_hash_map<std::pair<tsl::uint64 /*memory_space*/, OperationType>,
+  absl::flat_hash_map<std::pair<uint64_t /*memory_space*/, OperationType>,
                       OpMetrics_MemoryAccessed*>
       dst_memory_accessed_map;
   for (auto& dst_memory_accessed : *dst) {
@@ -108,7 +108,7 @@ void CombineMemoryAccessedBreakdown(
         &dst_memory_accessed;
   }
   for (const auto& src_memory_accessed : src) {
-    tsl::uint64 memory_space = src_memory_accessed.memory_space();
+    uint64_t memory_space = src_memory_accessed.memory_space();
     OperationType operation_type = src_memory_accessed.operation_type();
     auto*& dst_memory_accessed =
         dst_memory_accessed_map[{memory_space, operation_type}];

xprof/convert/op_stats_processor.cc

@@ -161,8 +161,8 @@ absl::Status OpStatsProcessor::Reduce(
         ParseHardwareType(op_stats_copy[i].run_environment().device_type()), i);
   }
 
-  StepIntersection step_intersection =
-      ComputeStepIntersectionToMergeOpStats(all_op_stats_info, kuint32max);
+  StepIntersection step_intersection = ComputeStepIntersectionToMergeOpStats(
+      all_op_stats_info, std::numeric_limits<uint32_t>::max());
   OpStats combined_op_stats;
   CombineAllOpStats(all_op_stats_info, step_intersection, &combined_op_stats);
 

xprof/convert/op_stats_to_pod_stats.cc

@@ -44,7 +44,7 @@ PodStatsRecord CreatePodStatsRecord(absl::string_view host_name,
   GenericStepBreakdown generic;
   bool success = step_info.step_breakdown().UnpackTo(&generic);
   DCHECK(success);
-  record.set_host_name(tsl::string(host_name));
+  record.set_host_name(host_name);
   record.set_step_num(step_info.step_num());
   record.set_total_duration_us(
       tsl::profiler::PicoToMicro(step_info.duration_ps()));

xprof/convert/xplane_to_op_metrics_db.cc

@@ -385,7 +385,7 @@ OpMetricsDb ConvertDeviceTraceXPlaneToOpMetricsDb(
   OpMetricsDb result;
   DeviceOpMetricsDbBuilder device_op_metrics_db_builder(&result);
 
-  int64_t first_op_offset_ps = tsl::kint64max;
+  int64_t first_op_offset_ps = std::numeric_limits<int64_t>::max();
   int64_t last_op_offset_ps = 0;
   int64_t num_tf_ops = 0;