Skip to content

Add option to use_umd driver#68

Open
broskoTT wants to merge 4 commits intotenstorrent:mainfrom
broskoTT:brosko/pyumd
Open

Add option to use_umd driver#68
broskoTT wants to merge 4 commits intotenstorrent:mainfrom
broskoTT:brosko/pyumd

Conversation

@broskoTT
Copy link

@broskoTT broskoTT commented Nov 3, 2025

Related issue: tenstorrent/tt-umd#1433
This PR is an introductory PR to UMD in tt-flash.

The change introduces --use_umd flag, which would turn on using UMD instead of pyluwen for all of the functionality.
tt-umd doesn't support Grayskull so this won't be available with UMD flag.

Changes:

  • Consume tt_umd package
  • Reset through umd
  • spi_read and spi_write through tt_umd
  • arc_msg, and bh way of getting fw version through tt_umd

Testing:
Tested on N300 and P150, to be retested on all configurations

There are many relevant UMD PRs to enable this, and this change will switch to main branch once those are merged to UMD main.

broskoTT added a commit to tenstorrent/tt-umd that referenced this pull request Jan 25, 2026
### Issue
#1431

### Description
This PR is an introductory PR to SPI. It defines the interface, how it's
used, how to construct TTDevice to be able to use it, and some tests. It
will remain non functional after this PR, and will be implemented per
arch in separate PRs.

### List of the changes
- Introduce spi.hpp
- Add allow_spi argument to TTDevice creation which will control whether
SPI access was enabled
- CPP and python tests which test the spi functionality.

### Testing
This functionality was extensively tested manually through the change
allowing flashing new FW to the chip through tt-flash:
tenstorrent/tt-flash#68
Also unit tests were carefully constructed.
Also, I executed code step by step to confirm everything matches luwen's
implementation.

### API Changes
Introduces spi read/write.
broskoTT added a commit to tenstorrent/tt-umd that referenced this pull request Jan 25, 2026
### Issue
#1431 

### Description
Implement SPI IO for Wormhole

### List of the changes
- Implement Wormhole SPI
- Minor change in wormhole constants

### Testing
This functionality was extensively tested manually through the change
allowing flashing new FW to the chip through tt-flash:
tenstorrent/tt-flash#68
Also unit tests were carefully constructed.
Also, I executed code step by step to confirm everything matches luwen's
implementation.

### API Changes
There are no API changes in this PR.
broskoTT added a commit to tenstorrent/tt-umd that referenced this pull request Feb 13, 2026
### Issue
#1431 

### Description
Implement SPI IO for Blackhole

### List of the changes
- Implement Blackhole SPI

### Testing
This functionality was extensively tested manually through the change
allowing flashing new FW to the chip through tt-flash:
tenstorrent/tt-flash#68
Also unit tests were carefully constructed.
Also, I executed code step by step to confirm everything matches luwen's
implementation.

### API Changes
There are no API changes in this PR.
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant