tinygrad

mirror of https://github.com/tinygrad/tinygrad.git synced 2026-04-29 03:00:14 -04:00

Files

qazal b7e3fbe07e llama: add VIZ=-1 to dev_run (#14583 )

* llama: add VIZ=-1 to dev_run

* readme

* cleaner

* add profile.sh script

* better grouping of options

* add other row

* readme edits

* work

2026-02-06 22:59:22 +09:00

cli.py

llama: add VIZ=-1 to dev_run (#14583 )

2026-02-06 22:59:22 +09:00

README

llama: add VIZ=-1 to dev_run (#14583 )

2026-02-06 22:59:22 +09:00

README

A command line tool for exploring the VIZ trace.

After running with VIZ=-1, use `PYTHONPATH=. extra/viz/cli.py` to explore the saved trace files.

## Inspect runtime profiling

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

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

## Inspect codegen and PatternMatcher

Use `PYTHONPATH=. 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 Program"`
Inspect a graph rewrite: `--rewrites --kernel E_3 --select "initial symbolic"`