compute-breakdown

Official

Analyze TPU pretraining compute efficiency and identify bottlenecks.

Authorprimatrix
Version1.0.0
Installs0

System Documentation

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'.

Dependency Matrix

Required Modules

xplaneprofile-anatomy

Components

scriptsreferences

💻 Claude Code Installation

Recommended: Let Claude install automatically. Simply copy and paste the text below to Claude Code.

Please help me install this Skill:
Name: compute-breakdown
Download link: https://github.com/primatrix/skills/archive/main.zip#compute-breakdown

Please download this .zip file, extract it, and install it in the .claude/skills/ directory.
View Source Repository

Agent Skills Search Helper

Install a tiny helper to your Agent, search and equip skill from 620,000+ vetted skills library on demand.