compute-breakdown

Analyze TPU pretraining compute efficiency from XSpace/XPlane profile data.

5|Updated Feb 4, 2026
One-click install
npx skills add https://github.com/primatrix/skills --skill compute-breakdown
Or copy as Structured Prompt for Agent
Please help me install this Agent Skill.
Skill: compute-breakdown
Source: https://github.com/primatrix/skills/tree/main/plugins/tpu-perf/skills/compute-breakdown
Command: npx skills add https://github.com/primatrix/skills --skill compute-breakdown

SYSTEM DOCUMENTATION & REQUIREMENTS

💡 This Skill requires xplane, profile-anatomy, and includes scripts (resource) and references (resource) components.

What problem does it solve?

This Skill analyzes TPU pretraining compute efficiency, providing insights into performance bottlenecks and resource utilization.

Core Features & Use Cases

  • Compute Efficiency Analysis: Break down compute time for TPU pretraining profiles, identifying time-consuming operations and layers.
  • Layer Scoping: Analyze compute time by layer, pinpointing performance issues within specific layers or modules.
  • Non-Compute Analysis: Identify non-compute operations such as data movement and communication, providing a comprehensive view of resource usage.
  • Roofline Analysis: Evaluate compute and memory utilization against theoretical peak performance, identifying opportunities for optimization.
  • Use Case: Imagine you have a TPU pretraining profile with a performance bottleneck. Use this Skill to analyze the profile, identify the root cause, and optimize the model or training process.

Quick Start

Use the compute-breakdown skill to analyze the compute efficiency of the TPU pretraining profile located at '/path/to/profile.xplane.pb'.

Frequently Asked Questions about compute-breakdown

High-intent search queries and answers about installing and using this skill.

FAQPage Schema
How do I analyze TPU pretraining compute efficiency from an XSpace profile?

To analyze TPU pretraining compute efficiency, process XSpace profile data to break down compute time, identify bottlenecks, and evaluate resource utilization. The skill parses XPlane and XLine data to pinpoint performance issues.

What is roofline analysis and how does it identify TPU performance bottlenecks?

Roofline analysis evaluates TPU compute and memory utilization against theoretical peak performance. By comparing actual operations to peak limits, it identifies bottlenecks and reveals opportunities for optimizing pretraining workloads.

How can I identify non-compute operations slowing down my TPU training?

To identify non-compute operations slowing TPU training, analyze the profile for data movement and communication overhead. The skill isolates these non-compute operations to provide a comprehensive view of resource usage beyond pure computation.

Do I need xplane parsing tools to run a layer-wise compute breakdown?

Yes, xplane parsing and analysis tools are required dependencies. The compute breakdown skill relies on these tools to process underlying XEvent and XStat data for accurate layer-wise performance scoping.

Can I pinpoint performance issues within specific layers of my TPU model?

Yes, you can pinpoint performance issues within specific layers using layer scoping. The skill analyzes compute time by layer or module, helping you locate exactly where performance bottlenecks occur during TPU pretraining.