.

tenstorrent · Feb 24, 2025 · 4be4c2f · 4be4c2f
1 parent b54bf3a
commit 4be4c2f
Show file tree

Hide file tree

Showing 8 changed files with 10 additions and 10 deletions.
diff --git a/.github/Dockerfile.ci b/.github/Dockerfile.ci
@@ -1,7 +1,7 @@
 ARG GIT_SHA
 ARG FROM_TAG=${GIT_SHA:-latest}
 
-FROM ghcr.io/tenstorrent/tt-torch/tt-torch-base-llvm-uplift-ubuntu-22-04:${FROM_TAG} AS ci-build
+FROM ghcr.io/tenstorrent/tt-torch/tt-torch-base-ubuntu-22-04:${FROM_TAG} AS ci-build
 SHELL ["/bin/bash", "-c"]
 
 # Create a directory for the build and toolchain
@@ -24,7 +24,7 @@ RUN cd third_party && \
     cmake -B toolchain -DBUILD_TOOLCHAIN=ON
 
 # Final stage
-FROM ghcr.io/tenstorrent/tt-torch/tt-torch-base-llvm-uplift-ubuntu-22-04:${FROM_TAG} AS ci
+FROM ghcr.io/tenstorrent/tt-torch/tt-torch-base-ubuntu-22-04:${FROM_TAG} AS ci
 
 # Copy the TTMLIR_TOOLCHAIN_DIR from the previous stage
 ENV TTMLIR_TOOLCHAIN_DIR=/opt/ttmlir-toolchain

diff --git a/.github/workflows/build-image.yml b/.github/workflows/build-image.yml
@@ -12,8 +12,8 @@ jobs:
       - wormhole_b0
 
     env:
-      BASE_IMAGE_NAME:      ghcr.io/${{ github.repository }}/tt-torch-base-llvm-uplift-ubuntu-22-04
-      CI_IMAGE_NAME:        ghcr.io/${{ github.repository }}/tt-torch-ci-llvm-uplift-ubuntu-22-04
+      BASE_IMAGE_NAME:      ghcr.io/${{ github.repository }}/tt-torch-base-ubuntu-22-04
+      CI_IMAGE_NAME:        ghcr.io/${{ github.repository }}/tt-torch-ci-ubuntu-22-04
 
     steps:
 

diff --git a/.github/workflows/generate-model-report.yml b/.github/workflows/generate-model-report.yml
@@ -17,7 +17,7 @@ jobs:
       - wormhole_b0
 
     container:
-      image: ghcr.io/tenstorrent/tt-torch/tt-torch-ci-llvm-uplift-ubuntu-22-04:latest
+      image: ghcr.io/tenstorrent/tt-torch/tt-torch-ci-ubuntu-22-04:latest
       options: --user root --device /dev/tenstorrent/0
       volumes:
         - /dev/hugepages:/dev/hugepages

diff --git a/.github/workflows/generate-ttnn-md.yml b/.github/workflows/generate-ttnn-md.yml
@@ -14,7 +14,7 @@ jobs:
     timeout-minutes: 120
     runs-on: ubuntu-latest
     container:
-      image: ghcr.io/tenstorrent/tt-torch/tt-torch-ci-llvm-uplift-ubuntu-22-04:latest
+      image: ghcr.io/tenstorrent/tt-torch/tt-torch-ci-ubuntu-22-04:latest
       options: --user root
 
     steps:

diff --git a/.github/workflows/run-build.yml b/.github/workflows/run-build.yml
@@ -22,7 +22,7 @@ jobs:
       - ${{ matrix.build.runs-on }}
 
     container:
-      image: ghcr.io/tenstorrent/tt-torch/tt-torch-ci-llvm-uplift-ubuntu-22-04:latest
+      image: ghcr.io/tenstorrent/tt-torch/tt-torch-ci-ubuntu-22-04:latest
       options: --user root --device /dev/tenstorrent/0
       volumes:
         - /dev/hugepages:/dev/hugepages

diff --git a/.github/workflows/run-e2e-tests.yml b/.github/workflows/run-e2e-tests.yml
@@ -21,7 +21,7 @@ jobs:
       - ${{ matrix.build.runs-on }}
 
     container:
-      image: ghcr.io/tenstorrent/tt-torch/tt-torch-ci-llvm-uplift-ubuntu-22-04:latest
+      image: ghcr.io/tenstorrent/tt-torch/tt-torch-ci-ubuntu-22-04:latest
       options: --user root --device /dev/tenstorrent/0 --shm-size=2gb
       volumes:
         - /dev/hugepages:/dev/hugepages

diff --git a/.github/workflows/run-op-by-op-model-tests.yml b/.github/workflows/run-op-by-op-model-tests.yml
@@ -218,7 +218,7 @@ jobs:
     name: "tests (${{ matrix.build.runs-on }}, ${{ matrix.build.name }})"
 
     container:
-      image: ghcr.io/tenstorrent/tt-torch/tt-torch-ci-llvm-uplift-ubuntu-22-04:latest
+      image: ghcr.io/tenstorrent/tt-torch/tt-torch-ci-ubuntu-22-04:latest
       options: --user root --device /dev/tenstorrent/0 --shm-size=4gb
       volumes:
         - /dev/hugepages:/dev/hugepages

diff --git a/.github/workflows/run-tests.yml b/.github/workflows/run-tests.yml
@@ -26,7 +26,7 @@ jobs:
       - ${{ matrix.build.runs-on }}
 
     container:
-      image: ghcr.io/tenstorrent/tt-torch/tt-torch-ci-llvm-uplift-ubuntu-22-04:latest
+      image: ghcr.io/tenstorrent/tt-torch/tt-torch-ci-ubuntu-22-04:latest
       options: --user root --device /dev/tenstorrent/0 --shm-size=2gb
       volumes:
         - /dev/hugepages:/dev/hugepages