Enable ROCm CI support. by akashveramd · Pull Request #1260 · pytorch/torchtitan

Commits on Sep 26, 2025

Added support to run torchtitan tests on ROCm.
akashveramd
committed
Added rocm ci support for integration_test_h100.
akashveramd
committed
Fixed a bug in build script. Removed ubuntu-cuda folder, instead using ubuntu folder for cuda Dockerfile.
akashveramd
committed
Added tests.integration_tests.features during rebase.
akashveramd
committed
Modified docker-builds.yml to build rocm docker image for torchtitan.
akashveramd
committed
Fixed runner for cuda and rocm images in docker-builds.yml.
akashveramd
committed
Added TEST_WITH_ROCM environment variable for running tests on rocm. Fixed error in integration_tests.py. Fixed lint errors.
akashveramd
committed
Made additional changes to tests.integration_tests.features during rebase.
akashveramd
committed
Changed runner to i-0962598bd0e8298b3 for building ROCm docker image.
akashveramd
committed
Changed runner to linux.12xlarge for building ROCm docker image.
akashveramd
committed
Changed runner to linux.2xlarge for building ROCm docker image.
akashveramd
committed
Resolved conflict in .github.workflows.integration_test_8gpu_models during rebase.
akashveramd
committed
Changed rocm docker image name in docker-builds.yml.
akashveramd
committed
Reverted the changes to integration_test_8gpu_h100.yaml.
akashveramd
committed
Empty dummy commit.
akashveramd
committed
Increased the timeout to 45 minutes to override timeout used in linux_job_v2.yml for integration_test_8gpu.yaml.
akashveramd
committed
Empty dummy commit.
akashveramd
committed
Added aws setup in the integration_test_8gpu workflow.
akashveramd
committed
Performed rebase and made changes to include code refactoring done upstream.
akashveramd
committed
Changed rocm runner name.
akashveramd
committed
Added a change to run build-test after aws-setup.
akashveramd
committed
Changed the test name in integration_test_8gpu.yaml workflow file.
akashveramd
committed

Commits on Sep 29, 2025

Commits on Sep 30, 2025

Moved permissions section for id-token outside the aws-setup job.
akashveramd
committed

Commits on Oct 6, 2025

Using move_aws_steps_inside_setup_rocm branch to do aws authentication in linux_job_v2.yml.
akashveramd
committed

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

Uh oh!

Enable ROCm CI support.#1260

Enable ROCm CI support.#1260
akashveramd wants to merge 29 commits into
pytorch:mainfrom
akashveramd:torchtitan_rocm_support_main

Commits on Sep 26, 2025

Added support to run torchtitan tests on ROCm.

Added rocm ci support for integration_test_h100.

Fixed a bug in build script. Removed ubuntu-cuda folder, instead using ubuntu folder for cuda Dockerfile.

Added tests.integration_tests.features during rebase.

Modified docker-builds.yml to build rocm docker image for torchtitan.

Fixed runner for cuda and rocm images in docker-builds.yml.

Added TEST_WITH_ROCM environment variable for running tests on rocm. Fixed error in integration_tests.py. Fixed lint errors.

Made additional changes to tests.integration_tests.features during rebase.

Changed runner to i-0962598bd0e8298b3 for building ROCm docker image.

Changed runner to linux.12xlarge for building ROCm docker image.

Changed runner to linux.2xlarge for building ROCm docker image.

Resolved conflict in .github.workflows.integration_test_8gpu_models during rebase.

Changed rocm docker image name in docker-builds.yml.

Reverted the changes to integration_test_8gpu_h100.yaml.

Empty dummy commit.

Increased the timeout to 45 minutes to override timeout used in linux_job_v2.yml for integration_test_8gpu.yaml.

Empty dummy commit.

Added aws setup in the integration_test_8gpu workflow.

Performed rebase and made changes to include code refactoring done upstream.

Changed rocm runner name.

Added a change to run build-test after aws-setup.

Changed the test name in integration_test_8gpu.yaml workflow file.

Commits on Sep 29, 2025

Fixed id-token permission issue in integration_test_8gpu.yaml.

Added id-token permission issue inside aws-setup job in integration_test_8gpu.yaml.

To test workflow, switched to 4 GPU runner as they are relatively easily available to run the workflow.

Commits on Sep 30, 2025

Moved permissions section for id-token outside the aws-setup job.

Commits on Oct 6, 2025

Using move_aws_steps_inside_setup_rocm branch to do aws authentication in linux_job_v2.yml.

Commits on Oct 7, 2025

Using linux linux_job_v2.yml in akashveramd fork having aws setup only for rocm.

Using linux_job_v2.yml having id-token write permissions from pytorch and move_aws_steps_inside_setup_rocm branch.