Test your own chip
The same tests run on an NVIDIA GPU, an Apple GPU or a CPU with one command. This page shows how to install and run them, and what is planned next.
- How to install the tester and run every probe on your own chip
- What is planned next on the roadmap
Run it on your chip
One command, ten result files, one pull request. Works on CUDA, Apple MPS and CPU today.
# install once, in its own virtual environment (Python 3.10 or newer) python3 -m venv chip-env && source chip-env/bin/activate pip install ai-chip-integrity # a one-minute smoke test: every probe and its self-test at small sizes chip-integrity run --quick # the full run: every probe and its self-test, ten OCP result files chip-integrity run --name NAME # or, on an NVIDIA GPU host with Docker, with nothing else installed docker run --rm --gpus all -v "$PWD:/results" ghcr.io/tech4biz-yasha/ai-chip-integrity:0.4.2 run --name NAME --out-dir /results
On many Macs the built-in python3 is version 3.9, which pip skips. Install a newer Python from python.org or with Homebrew first, then create the environment with its versioned command, for example python3.12 -m venv chip-env.
The same ten files come from running each probe by hand from a clone of the repository, for example python screen.py --out NAME_clean.jsonl and then python screen.py --inject 5 --out NAME_inject.jsonl for its self-test. The README lists all ten commands.
- Open a pull request with the ten files, the device name, driver version and PyTorch version.
- Rows are added only from attached files. The table is regenerated from them, so a row cannot say more than its file does.
- Found a fault? Open an issue with the file. A real finding on real hardware is the most useful thing anyone can contribute.
The output follows the OCP Test and Validation spec v2.0 through the official ocptv library, and the test suite validates it against the published schema.
Roadmap
Twelve probes, each aimed at one part of the chip. Six are built: 1, 2, 3, 6, 9 and 11.