From 3e549a66d988f8b6f6ced90db1412ca9ecb46286 Mon Sep 17 00:00:00 2001 From: Akash Verma Date: Thu, 12 Jun 2025 23:30:38 -0500 Subject: [PATCH] Modified docker-builds.yml to build rocm docker image for torchtitan. --- .github/workflows/docker-builds.yml | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) 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: