Skip to content

Commit

Permalink
Single Docker Image Release (#17051)
Browse files Browse the repository at this point in the history
### Ticket

#17040 

### Problem description
Since the ARCH_NAME dependency was removed, there is no longer a reason
to have multiple images.

### What's changed

- Change the workflows to generate only a single release image.
- Update the documentation
- Change paths in the Docker registry

### Checklist
- [x] Post commit CI passes
- [ ] Blackhole Post commit (if applicable)
- [ ] Model regression CI testing passes (if applicable)
- [ ] Device performance regression CI testing passes (if applicable)
- [ ] **(For models and ops writers)** Full [new
models](https://github.com/tenstorrent/tt-metal/actions/workflows/full-new-models-suite.yaml)
tests passes
- [ ] New/Existing tests provide coverage for changes
  • Loading branch information
dimitri-tenstorrent authored and patrickroberts committed Jan 25, 2025
1 parent fadd158 commit 8a093eb
Show file tree
Hide file tree
Showing 2 changed files with 6 additions and 16 deletions.
12 changes: 3 additions & 9 deletions .github/workflows/publish-release-image.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -19,7 +19,6 @@ jobs:
strategy:
matrix:
os: [ubuntu-20.04]
arch: [grayskull, wormhole_b0]
runs-on:
- build-docker
- in-service
Expand All @@ -42,7 +41,7 @@ jobs:
id: generate-tag-name
run: |
echo "WHEEL_FILENAME=$(ls -1 *.whl)" >> $GITHUB_ENV
REPO_IMAGE_NAME=ghcr.io/${{ github.repository }}/tt-metalium-${{ matrix.os }}-amd64-release/${{ matrix.arch }}
REPO_IMAGE_NAME=ghcr.io/${{ github.repository }}/tt-metalium-${{ matrix.os }}-amd64-release
echo "REPO_IMAGE_NAME=$REPO_IMAGE_NAME" >> $GITHUB_ENV
TAG_NAME=$REPO_IMAGE_NAME:${{ inputs.version }}
echo "TAG_NAME=$TAG_NAME" >> $GITHUB_ENV
Expand All @@ -64,23 +63,19 @@ jobs:
test_group:
[
{
arch: grayskull,
runs-on: ["cloud-virtual-machine", "E150", "in-service"],
cmd: pytest tests/end_to_end_tests,
},
{
arch: wormhole_b0,
runs-on: ["cloud-virtual-machine", "N150", "in-service"],
cmd: pytest tests/end_to_end_tests,
},
{
arch: wormhole_b0,
runs-on: ["cloud-virtual-machine", "N300", "in-service"],
cmd: pytest tests/end_to_end_tests,
},
]
env:
ARCH_NAME: ${{ matrix.test_group.arch }}
LOGURU_LEVEL: INFO
runs-on: ${{ matrix.test_group.runs-on }}
steps:
Expand All @@ -90,7 +85,7 @@ jobs:
timeout-minutes: ${{ inputs.timeout }}
uses: ./.github/actions/docker-run
with:
docker_os_arch: tt-metalium-${{ matrix.os }}-amd64-release/${{ matrix.test_group.arch }}
docker_os_arch: tt-metalium-${{ matrix.os }}-amd64-release
docker_version: ${{ inputs.version }}
docker_password: ${{ secrets.GITHUB_TOKEN }}
run_args: |
Expand All @@ -100,7 +95,6 @@ jobs:
strategy:
matrix:
os: [ubuntu-20.04]
arch: [grayskull, wormhole_b0]
runs-on:
- build-docker
- in-service
Expand All @@ -123,7 +117,7 @@ jobs:
LATEST_TAG=latest-rc
fi
echo "Determined that the current tag is " $LATEST_TAG
REPO_IMAGE_NAME=ghcr.io/${{ github.repository }}/tt-metalium-${{ matrix.os }}-amd64-release/${{ matrix.arch }}
REPO_IMAGE_NAME=ghcr.io/${{ github.repository }}/tt-metalium-${{ matrix.os }}-amd64-release
TAG_NAME=$REPO_IMAGE_NAME:${{ inputs.version }}
docker pull $TAG_NAME
echo "Tagging the image as " $REPO_IMAGE_NAME:$LATEST_TAG
Expand Down
10 changes: 3 additions & 7 deletions INSTALLING.md
Original file line number Diff line number Diff line change
Expand Up @@ -122,16 +122,12 @@ sudo cpupower frequency-set -g performance

#### Option 3: From Docker Release Image

Download the latest Docker release from our [Docker registry](https://github.com/orgs/tenstorrent/packages?q=tt-metalium-ubuntu&tab=packages&q=tt-metalium-ubuntu-20.04-amd64-release) page for
the particular Tenstorrent card architecture that you have installed on your
system. (ie. Grayskull, Wormhole, etc)
Download the latest Docker release from our [Docker registry](https://github.com/orgs/tenstorrent/packages?q=tt-metalium-ubuntu&tab=packages&q=tt-metalium-ubuntu-20.04-amd64-release) page

```sh
docker pull ghcr.io/tenstorrent/tt-metal/tt-metalium-ubuntu-20.04-amd64-release/<arch_name>:latest-rc
docker run -it --rm -v /dev/hugepages-1G:/dev/hugepages-1G --device /dev/tenstorrent ghcr.io/tenstorrent/tt-metal/tt-metalium-ubuntu-20.04-amd64-release/<arch_name>:latest-rc bash
docker pull ghcr.io/tenstorrent/tt-metal/tt-metalium-ubuntu-20.04-amd64-release:latest-rc
docker run -it --rm -v /dev/hugepages-1G:/dev/hugepages-1G --device /dev/tenstorrent ghcr.io/tenstorrent/tt-metal/tt-metalium-ubuntu-20.04-amd64-release:latest-rc bash
```
where `arch_name` is one of `grayskull`, `wormhole_b0`, or `blackhole`,
depending on your Tenstorrent card type.

When inside of the container,
```sh
Expand Down

0 comments on commit 8a093eb

Please sign in to comment.