[ROCm] upgrade ROCm CI builds to py3.10 (#134108)

Upgrade ROCm CI builds to py3.10

Pull Request resolved: https://github.com/pytorch/pytorch/pull/134108
Approved by: https://github.com/jeffdaily, https://github.com/jithunnair-amd, https://github.com/atalman
This commit is contained in:
Jack Taylor
2024-09-18 17:39:34 +00:00
committed by PyTorch MergeBot
parent 3efaa016b1
commit bad69044d8
7 changed files with 50 additions and 50 deletions

View File

@ -286,7 +286,7 @@ case "$image" in
TRITON=yes TRITON=yes
;; ;;
pytorch-linux-focal-rocm-n-1-py3) pytorch-linux-focal-rocm-n-1-py3)
ANACONDA_PYTHON_VERSION=3.8 ANACONDA_PYTHON_VERSION=3.10
GCC_VERSION=9 GCC_VERSION=9
PROTOBUF=yes PROTOBUF=yes
DB=yes DB=yes
@ -297,7 +297,7 @@ case "$image" in
TRITON=yes TRITON=yes
;; ;;
pytorch-linux-focal-rocm-n-py3) pytorch-linux-focal-rocm-n-py3)
ANACONDA_PYTHON_VERSION=3.8 ANACONDA_PYTHON_VERSION=3.10
GCC_VERSION=9 GCC_VERSION=9
PROTOBUF=yes PROTOBUF=yes
DB=yes DB=yes

View File

@ -31,13 +31,13 @@ jobs:
curr_branch: ${{ github.head_ref || github.ref_name }} curr_branch: ${{ github.head_ref || github.ref_name }}
curr_ref_type: ${{ github.ref_type }} curr_ref_type: ${{ github.ref_type }}
linux-focal-rocm6_1-py3_8-inductor-build: linux-focal-rocm6_1-py3_10-inductor-build:
name: rocm6.1-py3.8-inductor name: rocm6.1-py3.10-inductor
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-focal-rocm6.1-py3.8 build-environment: linux-focal-rocm6.1-py3.10
docker-image-name: pytorch-linux-focal-rocm-n-py3 docker-image-name: pytorch-linux-focal-rocm-n-py3
test-matrix: | test-matrix: |
{ include: [ { include: [
@ -45,14 +45,14 @@ jobs:
{ config: "inductor", shard: 2, num_shards: 2, runner: "linux.rocm.gpu.2" }, { config: "inductor", shard: 2, num_shards: 2, runner: "linux.rocm.gpu.2" },
]} ]}
linux-focal-rocm6_1-py3_8-inductor-test: linux-focal-rocm6_1-py3_10-inductor-test:
permissions: permissions:
id-token: write id-token: write
contents: read contents: read
name: rocm6.1-py3.8-inductor name: rocm6.1-py3.10-inductor
uses: ./.github/workflows/_rocm-test.yml uses: ./.github/workflows/_rocm-test.yml
needs: linux-focal-rocm6_1-py3_8-inductor-build needs: linux-focal-rocm6_1-py3_10-inductor-build
with: with:
build-environment: linux-focal-rocm6.1-py3.8 build-environment: linux-focal-rocm6.1-py3.10
docker-image: ${{ needs.linux-focal-rocm6_1-py3_8-inductor-build.outputs.docker-image }} docker-image: ${{ needs.linux-focal-rocm6_1-py3_10-inductor-build.outputs.docker-image }}
test-matrix: ${{ needs.linux-focal-rocm6_1-py3_8-inductor-build.outputs.test-matrix }} test-matrix: ${{ needs.linux-focal-rocm6_1-py3_10-inductor-build.outputs.test-matrix }}

View File

@ -297,13 +297,13 @@ jobs:
docker-image: ${{ needs.linux-vulkan-focal-py3_11-clang10-build.outputs.docker-image }} docker-image: ${{ needs.linux-vulkan-focal-py3_11-clang10-build.outputs.docker-image }}
test-matrix: ${{ needs.linux-vulkan-focal-py3_11-clang10-build.outputs.test-matrix }} test-matrix: ${{ needs.linux-vulkan-focal-py3_11-clang10-build.outputs.test-matrix }}
linux-focal-rocm6_1-py3_8-build: linux-focal-rocm6_1-py3_10-build:
name: linux-focal-rocm6.1-py3.8 name: linux-focal-rocm6.1-py3.10
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-focal-rocm6.1-py3.8 build-environment: linux-focal-rocm6.1-py3.10
docker-image-name: pytorch-linux-focal-rocm-n-py3 docker-image-name: pytorch-linux-focal-rocm-n-py3
test-matrix: | test-matrix: |
{ include: [ { include: [
@ -312,19 +312,19 @@ jobs:
{ config: "distributed", shard: 3, num_shards: 3, runner: "linux.rocm.gpu" }, { config: "distributed", shard: 3, num_shards: 3, runner: "linux.rocm.gpu" },
]} ]}
linux-focal-rocm6_1-py3_8-test: linux-focal-rocm6_1-py3_10-test:
permissions: permissions:
id-token: write id-token: write
contents: read contents: read
name: linux-focal-rocm6.1-py3.8 name: linux-focal-rocm6.1-py3.10
uses: ./.github/workflows/_rocm-test.yml uses: ./.github/workflows/_rocm-test.yml
needs: needs:
- linux-focal-rocm6_1-py3_8-build - linux-focal-rocm6_1-py3_10-build
- target-determination - target-determination
with: with:
build-environment: linux-focal-rocm6.1-py3.8 build-environment: linux-focal-rocm6.1-py3.10
docker-image: ${{ needs.linux-focal-rocm6_1-py3_8-build.outputs.docker-image }} docker-image: ${{ needs.linux-focal-rocm6_1-py3_10-build.outputs.docker-image }}
test-matrix: ${{ needs.linux-focal-rocm6_1-py3_8-build.outputs.test-matrix }} test-matrix: ${{ needs.linux-focal-rocm6_1-py3_10-build.outputs.test-matrix }}
linux-focal-cuda12_1-py3_10-gcc9-experimental-split-build: linux-focal-cuda12_1-py3_10-gcc9-experimental-split-build:
name: linux-focal-cuda12.1-py3.10-gcc9-experimental-split-build name: linux-focal-cuda12.1-py3.10-gcc9-experimental-split-build

View File

@ -503,15 +503,15 @@ jobs:
]} ]}
secrets: inherit secrets: inherit
linux-focal-rocm6_1-py3_8-build: linux-focal-rocm6_1-py3_10-build:
# don't run build twice on main # don't run build twice on main
if: github.event_name == 'pull_request' if: github.event_name == 'pull_request'
name: linux-focal-rocm6.1-py3.8 name: linux-focal-rocm6.1-py3.10
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-focal-rocm6.1-py3.8 build-environment: linux-focal-rocm6.1-py3.10
docker-image-name: pytorch-linux-focal-rocm-n-py3 docker-image-name: pytorch-linux-focal-rocm-n-py3
sync-tag: rocm-build sync-tag: rocm-build
test-matrix: | test-matrix: |

View File

@ -25,11 +25,11 @@ jobs:
id-token: write id-token: write
contents: read contents: read
linux-focal-rocm6_1-py3_8-build: linux-focal-rocm6_1-py3_10-build:
name: linux-focal-rocm6.1-py3.8 name: linux-focal-rocm6.1-py3.10
uses: ./.github/workflows/_linux-build.yml uses: ./.github/workflows/_linux-build.yml
with: with:
build-environment: linux-focal-rocm6.1-py3.8 build-environment: linux-focal-rocm6.1-py3.10
docker-image-name: pytorch-linux-focal-rocm-n-py3 docker-image-name: pytorch-linux-focal-rocm-n-py3
sync-tag: rocm-build sync-tag: rocm-build
test-matrix: | test-matrix: |
@ -42,16 +42,16 @@ jobs:
{ config: "default", shard: 6, num_shards: 6, runner: "linux.rocm.gpu.2" }, { config: "default", shard: 6, num_shards: 6, runner: "linux.rocm.gpu.2" },
]} ]}
linux-focal-rocm6_1-py3_8-test: linux-focal-rocm6_1-py3_10-test:
permissions: permissions:
id-token: write id-token: write
contents: read contents: read
name: linux-focal-rocm6.1-py3.8 name: linux-focal-rocm6.1-py3.10
uses: ./.github/workflows/_rocm-test.yml uses: ./.github/workflows/_rocm-test.yml
needs: needs:
- linux-focal-rocm6_1-py3_8-build - linux-focal-rocm6_1-py3_10-build
- target-determination - target-determination
with: with:
build-environment: linux-focal-rocm6.1-py3.8 build-environment: linux-focal-rocm6.1-py3.10
docker-image: ${{ needs.linux-focal-rocm6_1-py3_8-build.outputs.docker-image }} docker-image: ${{ needs.linux-focal-rocm6_1-py3_10-build.outputs.docker-image }}
test-matrix: ${{ needs.linux-focal-rocm6_1-py3_8-build.outputs.test-matrix }} test-matrix: ${{ needs.linux-focal-rocm6_1-py3_10-build.outputs.test-matrix }}

View File

@ -130,13 +130,13 @@ jobs:
docker-image: ${{ needs.linux-focal-py3_9-clang10-build.outputs.docker-image }} docker-image: ${{ needs.linux-focal-py3_9-clang10-build.outputs.docker-image }}
test-matrix: ${{ needs.linux-focal-py3_9-clang10-build.outputs.test-matrix }} test-matrix: ${{ needs.linux-focal-py3_9-clang10-build.outputs.test-matrix }}
linux-focal-rocm6_1-py3_8-build: linux-focal-rocm6_1-py3_10-build:
name: linux-focal-rocm6.1-py3.8 name: linux-focal-rocm6.1-py3.10
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-focal-rocm6.1-py3.8 build-environment: linux-focal-rocm6.1-py3.10
docker-image-name: pytorch-linux-focal-rocm-n-py3 docker-image-name: pytorch-linux-focal-rocm-n-py3
test-matrix: | test-matrix: |
{ include: [ { include: [
@ -144,19 +144,19 @@ jobs:
{ config: "slow", shard: 2, num_shards: 2, runner: "linux.rocm.gpu" }, { config: "slow", shard: 2, num_shards: 2, runner: "linux.rocm.gpu" },
]} ]}
linux-focal-rocm6_1-py3_8-test: linux-focal-rocm6_1-py3_10-test:
permissions: permissions:
id-token: write id-token: write
contents: read contents: read
name: linux-focal-rocm6.1-py3.8 name: linux-focal-rocm6.1-py3.10
uses: ./.github/workflows/_rocm-test.yml uses: ./.github/workflows/_rocm-test.yml
needs: needs:
- linux-focal-rocm6_1-py3_8-build - linux-focal-rocm6_1-py3_10-build
- target-determination - target-determination
with: with:
build-environment: linux-focal-rocm6.1-py3.8 build-environment: linux-focal-rocm6.1-py3.10
docker-image: ${{ needs.linux-focal-rocm6_1-py3_8-build.outputs.docker-image }} docker-image: ${{ needs.linux-focal-rocm6_1-py3_10-build.outputs.docker-image }}
test-matrix: ${{ needs.linux-focal-rocm6_1-py3_8-build.outputs.test-matrix }} test-matrix: ${{ needs.linux-focal-rocm6_1-py3_10-build.outputs.test-matrix }}
linux-jammy-py3_10-clang15-asan-build: linux-jammy-py3_10-clang15-asan-build:
name: linux-jammy-py3.10-clang15-asan name: linux-jammy-py3.10-clang15-asan

View File

@ -223,13 +223,13 @@ jobs:
cuda-version: "12.1" cuda-version: "12.1"
runner: "${{ needs.get-label-type.outputs.label-type }}windows.4xlarge.nonephemeral" runner: "${{ needs.get-label-type.outputs.label-type }}windows.4xlarge.nonephemeral"
linux-focal-rocm6_1-py3_8-build: linux-focal-rocm6_1-py3_10-build:
name: linux-focal-rocm6.1-py3.8 name: linux-focal-rocm6.1-py3.10
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-focal-rocm6.1-py3.8 build-environment: linux-focal-rocm6.1-py3.10
docker-image-name: pytorch-linux-focal-rocm-n-py3 docker-image-name: pytorch-linux-focal-rocm-n-py3
sync-tag: rocm-build sync-tag: rocm-build
test-matrix: | test-matrix: |
@ -240,19 +240,19 @@ jobs:
]} ]}
secrets: inherit secrets: inherit
linux-focal-rocm6_1-py3_8-test: linux-focal-rocm6_1-py3_10-test:
permissions: permissions:
id-token: write id-token: write
contents: read contents: read
name: linux-focal-rocm6.1-py3.8 name: linux-focal-rocm6.1-py3.10
uses: ./.github/workflows/_rocm-test.yml uses: ./.github/workflows/_rocm-test.yml
needs: needs:
- linux-focal-rocm6_1-py3_8-build - linux-focal-rocm6_1-py3_10-build
- target-determination - target-determination
with: with:
build-environment: linux-focal-rocm6.1-py3.8 build-environment: linux-focal-rocm6.1-py3.10
docker-image: ${{ needs.linux-focal-rocm6_1-py3_8-build.outputs.docker-image }} docker-image: ${{ needs.linux-focal-rocm6_1-py3_10-build.outputs.docker-image }}
test-matrix: ${{ needs.linux-focal-rocm6_1-py3_8-build.outputs.test-matrix }} test-matrix: ${{ needs.linux-focal-rocm6_1-py3_10-build.outputs.test-matrix }}
tests-to-include: "test_nn test_torch test_cuda test_ops test_unary_ufuncs test_binary_ufuncs test_autograd inductor/test_torchinductor distributed/test_c10d_common distributed/test_c10d_nccl" tests-to-include: "test_nn test_torch test_cuda test_ops test_unary_ufuncs test_binary_ufuncs test_autograd inductor/test_torchinductor distributed/test_c10d_common distributed/test_c10d_nccl"
linux-focal-cuda12_4-py3_10-gcc9-experimental-split-build: linux-focal-cuda12_4-py3_10-gcc9-experimental-split-build: