tinygrad

mirror of https://github.com/tinygrad/tinygrad.git synced 2026-06-15 09:33:03 +08:00

Files

qazal ec5b7a249e viz: refactor sqtt timeline builder (#15494 )

* viz: refactor sqtt timeline builder

* barrier maps to waves

* clean up cli

2026-03-26 21:16:15 +09:00

cli.py

viz: refactor sqtt timeline builder (#15494 )

2026-03-26 21:16:15 +09:00

README

viz/cli: remove --offset and --limit flags (#15439 )

2026-03-25 09:52:27 +09:00

README

A command line tool for exploring the VIZ trace.

# Lightweight tracing

Supported on all backends.

Flags: VIZ=-1 to only save the trace to a file, VIZ=1 also launches a web server.

1. Set VIZ to -1 to save the trace.
2. Use `extra/viz/cli.py` to inspect the trace files.

## Inspect runtime profiling

Use `extra/viz/cli.py --profile` to list all traced devices.

List top slowest kernels on a device: `--profile --device "AMD" | head 10`
List samples of a kernel on a device: `--profile --device "AMD" --kernel E_3 | head 4`

## Inspect codegen and PatternMatcher

Use `extra/viz/cli.py --rewrites` to list all traced kernels.

List all codegen steps for a kernel: `--rewrites --kernel E_3`
Get source code: `--rewrites --kernel E_3 --select "View Source"`
Inspect a graph rewrite: `--rewrites --kernel E_3 --select "initial symbolic"`

# SQTT tracing

Supported on AMD for RDNA3 and RDNA4 (best) and CDNA (developing).

Flags: VIZ=-2 to save SQTT trace to a file. VIZ=2 also launches a web server. View other flags in tinygrad/runtime/ops_amd.py to configure SQTT as needed.

Use `extra/viz/cli.py --profile | grep SQTT` to view all available SQTT traces.
You can select a specific trace option with --device, Example workflow:

```bash
# Run amd_asm_matmul with VIZ=-2 to capture the trace
VIZ=-2 python extra/gemm/amd_asm_matmul.py

# View barriers
extra/viz/cli.py --profile --device "Exec kernel SQTT PKTS SE:0" | rg BARRIER | head -10
```