diff --git a/.github/workflows/docker-builds.yml b/.github/workflows/docker-builds.yml index 11ff5390cc..75d5082c53 100644 --- a/.github/workflows/docker-builds.yml +++ b/.github/workflows/docker-builds.yml @@ -22,13 +22,16 @@ concurrency: jobs: docker-build: - runs-on: [self-hosted, linux.2xlarge] + runs-on: [self-hosted, linux.2xlarge, linux.rocm.gpu.mi300.8] timeout-minutes: 240 strategy: fail-fast: false matrix: include: - - docker-image-name: torchtitan-ubuntu-20.04-clang12 + - docker-image-name: [ + torchtitan-ubuntu-20.04-clang12, + torchtitan-rocm-pytorch-nightly-ubuntu-22.04-clang19-py3 + ] env: DOCKER_IMAGE: 308535385114.dkr.ecr.us-east-1.amazonaws.com/torchtitan/${{ matrix.docker-image-name }} steps: