Feat(tpu-info): Add support for buffer_transfer_latency metric.

PiperOrigin-RevId: 786195872

Author: Googler

tpu_info/tpu_info/cli_helper.py

@@ -146,6 +146,7 @@ def get_metric_table(
  metric_functions = {
  "hbm_usage": lambda: get_hbm_usage_table(chip_type, count),
  "duty_cycle_percent": lambda: get_duty_cycle_table(chip_type, count),
+ "buffer_transfer_latency": lambda: [BufferTransferLatencyTable().render()],
  }
  renderables.extend(metric_functions[metric_name]())
  return renderables
@@ -224,7 +225,6 @@ def get_duty_cycle_table(
  renderables.append(table)
  return renderables
 
-
 def get_device_usage(
  chip_type: device.TpuChip,
 ) -> List[metrics.Usage] | panel.Panel: