Skip to content

Latest commit

 

History

History
29 lines (14 loc) · 708 Bytes

README.silicon.md

File metadata and controls

29 lines (14 loc) · 708 Bytes
Raw

How to run pybuda on silicon

Docker

To create a docker, run bin/run-docker.sh from third_party/budabackend. Then, enter it use docker exec -u $USER -it special-$USER bash.

Env

Grayskull, Wormhole and Blackhole machines require slightly different setups, so there are setup scripts for each. To build everything run the appropriate script:

  • source env_for_silicon.sh (Grayskull)

or

  • source env_for_wormhole_b0.sh (Wormhole B0)

or

  • source env_for_blackhole.sh (Blackhole)

Now, you should be able to run pybuda pytests and python scripts.

Run

For example, try:

pytest -svv pybuda/test/backend/test_silicon.py::test_basic_training[Grayskull-acc1-mbc1-microbatch1-s1]