OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-06 00:20:18 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity

[ROCm][CI] Update build-environment for mi300 workflows (#153134)

Browse Source 
so their test times are tracked separately in https://raw.githubusercontent.com/pytorch/test-infra/generated-stats/stats/test-times.json. Currently, both MI200 and MI300 test times get combined into the same key `linux-focal-rocm-py3.10`

Pull Request resolved: https://github.com/pytorch/pytorch/pull/153134
Approved by: https://github.com/huydhn
This commit is contained in:
Jithun Nair 2025-05-28 19:04:53 +00:00 committed by PyTorch MergeBot
parent d4ab8e74f3
commit 24980d2641
3 changed files with 12 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
.github/workflows/inductor-rocm-mi300.yml vendored
View File

@ -38,12 +38,12 @@ jobs:
opt_out_experiments: lf opt_out_experiments: lf
linux-jammy-rocm-py3_10-inductor-build: linux-jammy-rocm-py3_10-inductor-build:
name: rocm-py3.10-inductor name: rocm-py3.10-inductor-mi300
uses: ./.github/workflows/_linux-build.yml uses: ./.github/workflows/_linux-build.yml
needs: get-label-type needs: get-label-type
with: with:
runner_prefix: "${{ needs.get-label-type.outputs.label-type }}" runner_prefix: "${{ needs.get-label-type.outputs.label-type }}"
build-environment: linux-jammy-rocm-py3.10 build-environment: linux-jammy-rocm-py3.10-mi300
docker-image-name: ci-image:pytorch-linux-jammy-rocm-n-py3 docker-image-name: ci-image:pytorch-linux-jammy-rocm-n-py3
test-matrix: | test-matrix: |
{ include: [ { include: [
@ -56,11 +56,11 @@ jobs:
permissions: permissions:
id-token: write id-token: write
contents: read contents: read
name: rocm-py3.10-inductor name: rocm-py3.10-inductor-mi300
uses: ./.github/workflows/_rocm-test.yml uses: ./.github/workflows/_rocm-test.yml
needs: linux-jammy-rocm-py3_10-inductor-build needs: linux-jammy-rocm-py3_10-inductor-build
with: with:
build-environment: linux-jammy-rocm-py3.10 build-environment: linux-jammy-rocm-py3.10-mi300
docker-image: ${{ needs.linux-jammy-rocm-py3_10-inductor-build.outputs.docker-image }} docker-image: ${{ needs.linux-jammy-rocm-py3_10-inductor-build.outputs.docker-image }}
test-matrix: ${{ needs.linux-jammy-rocm-py3_10-inductor-build.outputs.test-matrix }} test-matrix: ${{ needs.linux-jammy-rocm-py3_10-inductor-build.outputs.test-matrix }}
secrets: inherit secrets: inherit

8
.github/workflows/periodic-rocm-mi300.yml vendored
View File

@ -50,12 +50,12 @@ jobs:
curr_ref_type: ${{ github.ref_type }} curr_ref_type: ${{ github.ref_type }}
linux-jammy-rocm-py3_10-build: linux-jammy-rocm-py3_10-build:
name: linux-jammy-rocm-py3.10 name: linux-jammy-rocm-py3.10-mi300
uses: ./.github/workflows/_linux-build.yml uses: ./.github/workflows/_linux-build.yml
needs: get-label-type needs: get-label-type
with: with:
runner_prefix: "${{ needs.get-label-type.outputs.label-type }}" runner_prefix: "${{ needs.get-label-type.outputs.label-type }}"
build-environment: linux-jammy-rocm-py3.10 build-environment: linux-jammy-rocm-py3.10-mi300
docker-image-name: ci-image:pytorch-linux-jammy-rocm-n-py3 docker-image-name: ci-image:pytorch-linux-jammy-rocm-n-py3
test-matrix: | test-matrix: |
{ include: [ { include: [
@ -69,13 +69,13 @@ jobs:
permissions: permissions:
id-token: write id-token: write
contents: read contents: read
name: linux-jammy-rocm-py3.10 name: linux-jammy-rocm-py3.10-mi300
uses: ./.github/workflows/_rocm-test.yml uses: ./.github/workflows/_rocm-test.yml
needs: needs:
- linux-jammy-rocm-py3_10-build - linux-jammy-rocm-py3_10-build
- target-determination - target-determination
with: with:
build-environment: linux-jammy-rocm-py3.10 build-environment: linux-jammy-rocm-py3.10-mi300
docker-image: ${{ needs.linux-jammy-rocm-py3_10-build.outputs.docker-image }} docker-image: ${{ needs.linux-jammy-rocm-py3_10-build.outputs.docker-image }}
test-matrix: ${{ needs.linux-jammy-rocm-py3_10-build.outputs.test-matrix }} test-matrix: ${{ needs.linux-jammy-rocm-py3_10-build.outputs.test-matrix }}
secrets: inherit secrets: inherit

8
.github/workflows/rocm-mi300.yml vendored
View File

@ -38,12 +38,12 @@ jobs:
linux-jammy-rocm-py3_10-build: linux-jammy-rocm-py3_10-build:
if: ${{ (github.event_name != 'schedule' || github.repository == 'pytorch/pytorch') && github.repository_owner == 'pytorch' }} if: ${{ (github.event_name != 'schedule' || github.repository == 'pytorch/pytorch') && github.repository_owner == 'pytorch' }}
name: linux-jammy-rocm-py3.10 name: linux-jammy-rocm-py3.10-mi300
uses: ./.github/workflows/_linux-build.yml uses: ./.github/workflows/_linux-build.yml
needs: get-label-type needs: get-label-type
with: with:
runner_prefix: "${{ needs.get-label-type.outputs.label-type }}" runner_prefix: "${{ needs.get-label-type.outputs.label-type }}"
build-environment: linux-jammy-rocm-py3.10 build-environment: linux-jammy-rocm-py3.10-mi300
docker-image-name: ci-image:pytorch-linux-jammy-rocm-n-py3 docker-image-name: ci-image:pytorch-linux-jammy-rocm-n-py3
sync-tag: rocm-build sync-tag: rocm-build
test-matrix: | test-matrix: |
@ -61,13 +61,13 @@ jobs:
permissions: permissions:
id-token: write id-token: write
contents: read contents: read
name: linux-jammy-rocm-py3.10 name: linux-jammy-rocm-py3.10-mi300
uses: ./.github/workflows/_rocm-test.yml uses: ./.github/workflows/_rocm-test.yml
needs: needs:
- linux-jammy-rocm-py3_10-build - linux-jammy-rocm-py3_10-build
- target-determination - target-determination
with: with:
build-environment: linux-jammy-rocm-py3.10 build-environment: linux-jammy-rocm-py3.10-mi300
docker-image: ${{ needs.linux-jammy-rocm-py3_10-build.outputs.docker-image }} docker-image: ${{ needs.linux-jammy-rocm-py3_10-build.outputs.docker-image }}
test-matrix: ${{ needs.linux-jammy-rocm-py3_10-build.outputs.test-matrix }} test-matrix: ${{ needs.linux-jammy-rocm-py3_10-build.outputs.test-matrix }}
secrets: inherit secrets: inherit