Skip to content

GitLab

Unverified Commit 87c5e572 authored by KyleCZH's avatar KyleCZH Committed by GitHub
Browse files

add rocm4.3.1 to nightly wheel builds (#4516)

parent b81d189d
Hide whitespace changes
Inline Side-by-side
Showing with 85 additions and 85 deletions
.circleci/config.yml
View file @ 87c5e572
...@@ -1008,16 +1008,16 @@ workflows: ...@@ -1008,16 +1008,16 @@ workflows:
name: binary_linux_wheel_py3.6_cu113 name: binary_linux_wheel_py3.6_cu113
python_version: '3.6' python_version: '3.6'
wheel_docker_image: pytorch/manylinux-cuda113 wheel_docker_image: pytorch/manylinux-cuda113
- binary_linux_wheel:
cu_version: rocm4.1
name: binary_linux_wheel_py3.6_rocm4.1
python_version: '3.6'
wheel_docker_image: pytorch/manylinux-rocm:4.1
- binary_linux_wheel: - binary_linux_wheel:
cu_version: rocm4.2 cu_version: rocm4.2
name: binary_linux_wheel_py3.6_rocm4.2 name: binary_linux_wheel_py3.6_rocm4.2
python_version: '3.6' python_version: '3.6'
wheel_docker_image: pytorch/manylinux-rocm:4.2 wheel_docker_image: pytorch/manylinux-rocm:4.2
- binary_linux_wheel:
cu_version: rocm4.3.1
name: binary_linux_wheel_py3.6_rocm4.3.1
python_version: '3.6'
wheel_docker_image: pytorch/manylinux-rocm:4.3.1
- binary_linux_wheel: - binary_linux_wheel:
conda_docker_image: pytorch/conda-builder:cpu conda_docker_image: pytorch/conda-builder:cpu
cu_version: cpu cu_version: cpu
...@@ -1047,16 +1047,16 @@ workflows: ...@@ -1047,16 +1047,16 @@ workflows:
name: binary_linux_wheel_py3.7_cu113 name: binary_linux_wheel_py3.7_cu113
python_version: '3.7' python_version: '3.7'
wheel_docker_image: pytorch/manylinux-cuda113 wheel_docker_image: pytorch/manylinux-cuda113
- binary_linux_wheel:
cu_version: rocm4.1
name: binary_linux_wheel_py3.7_rocm4.1
python_version: '3.7'
wheel_docker_image: pytorch/manylinux-rocm:4.1
- binary_linux_wheel: - binary_linux_wheel:
cu_version: rocm4.2 cu_version: rocm4.2
name: binary_linux_wheel_py3.7_rocm4.2 name: binary_linux_wheel_py3.7_rocm4.2
python_version: '3.7' python_version: '3.7'
wheel_docker_image: pytorch/manylinux-rocm:4.2 wheel_docker_image: pytorch/manylinux-rocm:4.2
- binary_linux_wheel:
cu_version: rocm4.3.1
name: binary_linux_wheel_py3.7_rocm4.3.1
python_version: '3.7'
wheel_docker_image: pytorch/manylinux-rocm:4.3.1
- binary_linux_wheel: - binary_linux_wheel:
conda_docker_image: pytorch/conda-builder:cpu conda_docker_image: pytorch/conda-builder:cpu
cu_version: cpu cu_version: cpu
...@@ -1081,16 +1081,16 @@ workflows: ...@@ -1081,16 +1081,16 @@ workflows:
name: binary_linux_wheel_py3.8_cu113 name: binary_linux_wheel_py3.8_cu113
python_version: '3.8' python_version: '3.8'
wheel_docker_image: pytorch/manylinux-cuda113 wheel_docker_image: pytorch/manylinux-cuda113
- binary_linux_wheel:
cu_version: rocm4.1
name: binary_linux_wheel_py3.8_rocm4.1
python_version: '3.8'
wheel_docker_image: pytorch/manylinux-rocm:4.1
- binary_linux_wheel: - binary_linux_wheel:
cu_version: rocm4.2 cu_version: rocm4.2
name: binary_linux_wheel_py3.8_rocm4.2 name: binary_linux_wheel_py3.8_rocm4.2
python_version: '3.8' python_version: '3.8'
wheel_docker_image: pytorch/manylinux-rocm:4.2 wheel_docker_image: pytorch/manylinux-rocm:4.2
- binary_linux_wheel:
cu_version: rocm4.3.1
name: binary_linux_wheel_py3.8_rocm4.3.1
python_version: '3.8'
wheel_docker_image: pytorch/manylinux-rocm:4.3.1
- binary_linux_wheel: - binary_linux_wheel:
conda_docker_image: pytorch/conda-builder:cpu conda_docker_image: pytorch/conda-builder:cpu
cu_version: cpu cu_version: cpu
...@@ -1115,16 +1115,16 @@ workflows: ...@@ -1115,16 +1115,16 @@ workflows:
name: binary_linux_wheel_py3.9_cu113 name: binary_linux_wheel_py3.9_cu113
python_version: '3.9' python_version: '3.9'
wheel_docker_image: pytorch/manylinux-cuda113 wheel_docker_image: pytorch/manylinux-cuda113
- binary_linux_wheel:
cu_version: rocm4.1
name: binary_linux_wheel_py3.9_rocm4.1
python_version: '3.9'
wheel_docker_image: pytorch/manylinux-rocm:4.1
- binary_linux_wheel: - binary_linux_wheel:
cu_version: rocm4.2 cu_version: rocm4.2
name: binary_linux_wheel_py3.9_rocm4.2 name: binary_linux_wheel_py3.9_rocm4.2
python_version: '3.9' python_version: '3.9'
wheel_docker_image: pytorch/manylinux-rocm:4.2 wheel_docker_image: pytorch/manylinux-rocm:4.2
- binary_linux_wheel:
cu_version: rocm4.3.1
name: binary_linux_wheel_py3.9_rocm4.3.1
python_version: '3.9'
wheel_docker_image: pytorch/manylinux-rocm:4.3.1
- binary_macos_wheel: - binary_macos_wheel:
conda_docker_image: pytorch/conda-builder:cpu conda_docker_image: pytorch/conda-builder:cpu
cu_version: cpu cu_version: cpu
...@@ -1887,15 +1887,15 @@ workflows: ...@@ -1887,15 +1887,15 @@ workflows:
requires: requires:
- nightly_binary_linux_wheel_py3.6_cu113_upload - nightly_binary_linux_wheel_py3.6_cu113_upload
- binary_linux_wheel: - binary_linux_wheel:
cu_version: rocm4.1 cu_version: rocm4.2
filters: filters:
branches: branches:
only: nightly only: nightly
tags: tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.6_rocm4.1 name: nightly_binary_linux_wheel_py3.6_rocm4.2
python_version: '3.6' python_version: '3.6'
wheel_docker_image: pytorch/manylinux-rocm:4.1 wheel_docker_image: pytorch/manylinux-rocm:4.2
- binary_wheel_upload: - binary_wheel_upload:
context: org-member context: org-member
filters: filters:
...@@ -1903,29 +1903,29 @@ workflows: ...@@ -1903,29 +1903,29 @@ workflows:
only: nightly only: nightly
tags: tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.6_rocm4.1_upload name: nightly_binary_linux_wheel_py3.6_rocm4.2_upload
requires: requires:
- nightly_binary_linux_wheel_py3.6_rocm4.1 - nightly_binary_linux_wheel_py3.6_rocm4.2
subfolder: rocm4.1/ subfolder: rocm4.2/
- smoke_test_linux_pip: - smoke_test_linux_pip:
filters: filters:
branches: branches:
only: only:
- nightly - nightly
name: nightly_binary_linux_wheel_py3.6_rocm4.1_smoke_test_pip name: nightly_binary_linux_wheel_py3.6_rocm4.2_smoke_test_pip
python_version: '3.6' python_version: '3.6'
requires: requires:
- nightly_binary_linux_wheel_py3.6_rocm4.1_upload - nightly_binary_linux_wheel_py3.6_rocm4.2_upload
- binary_linux_wheel: - binary_linux_wheel:
cu_version: rocm4.2 cu_version: rocm4.3.1
filters: filters:
branches: branches:
only: nightly only: nightly
tags: tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.6_rocm4.2 name: nightly_binary_linux_wheel_py3.6_rocm4.3.1
python_version: '3.6' python_version: '3.6'
wheel_docker_image: pytorch/manylinux-rocm:4.2 wheel_docker_image: pytorch/manylinux-rocm:4.3.1
- binary_wheel_upload: - binary_wheel_upload:
context: org-member context: org-member
filters: filters:
...@@ -1933,19 +1933,19 @@ workflows: ...@@ -1933,19 +1933,19 @@ workflows:
only: nightly only: nightly
tags: tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.6_rocm4.2_upload name: nightly_binary_linux_wheel_py3.6_rocm4.3.1_upload
requires: requires:
- nightly_binary_linux_wheel_py3.6_rocm4.2 - nightly_binary_linux_wheel_py3.6_rocm4.3.1
subfolder: rocm4.2/ subfolder: rocm4.3.1/
- smoke_test_linux_pip: - smoke_test_linux_pip:
filters: filters:
branches: branches:
only: only:
- nightly - nightly
name: nightly_binary_linux_wheel_py3.6_rocm4.2_smoke_test_pip name: nightly_binary_linux_wheel_py3.6_rocm4.3.1_smoke_test_pip
python_version: '3.6' python_version: '3.6'
requires: requires:
- nightly_binary_linux_wheel_py3.6_rocm4.2_upload - nightly_binary_linux_wheel_py3.6_rocm4.3.1_upload
- binary_linux_wheel: - binary_linux_wheel:
conda_docker_image: pytorch/conda-builder:cpu conda_docker_image: pytorch/conda-builder:cpu
cu_version: cpu cu_version: cpu
...@@ -2071,15 +2071,15 @@ workflows: ...@@ -2071,15 +2071,15 @@ workflows:
requires: requires:
- nightly_binary_linux_wheel_py3.7_cu113_upload - nightly_binary_linux_wheel_py3.7_cu113_upload
- binary_linux_wheel: - binary_linux_wheel:
cu_version: rocm4.1 cu_version: rocm4.2
filters: filters:
branches: branches:
only: nightly only: nightly
tags: tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.7_rocm4.1 name: nightly_binary_linux_wheel_py3.7_rocm4.2
python_version: '3.7' python_version: '3.7'
wheel_docker_image: pytorch/manylinux-rocm:4.1 wheel_docker_image: pytorch/manylinux-rocm:4.2
- binary_wheel_upload: - binary_wheel_upload:
context: org-member context: org-member
filters: filters:
...@@ -2087,29 +2087,29 @@ workflows: ...@@ -2087,29 +2087,29 @@ workflows:
only: nightly only: nightly
tags: tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.7_rocm4.1_upload name: nightly_binary_linux_wheel_py3.7_rocm4.2_upload
requires: requires:
- nightly_binary_linux_wheel_py3.7_rocm4.1 - nightly_binary_linux_wheel_py3.7_rocm4.2
subfolder: rocm4.1/ subfolder: rocm4.2/
- smoke_test_linux_pip: - smoke_test_linux_pip:
filters: filters:
branches: branches:
only: only:
- nightly - nightly
name: nightly_binary_linux_wheel_py3.7_rocm4.1_smoke_test_pip name: nightly_binary_linux_wheel_py3.7_rocm4.2_smoke_test_pip
python_version: '3.7' python_version: '3.7'
requires: requires:
- nightly_binary_linux_wheel_py3.7_rocm4.1_upload - nightly_binary_linux_wheel_py3.7_rocm4.2_upload
- binary_linux_wheel: - binary_linux_wheel:
cu_version: rocm4.2 cu_version: rocm4.3.1
filters: filters:
branches: branches:
only: nightly only: nightly
tags: tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.7_rocm4.2 name: nightly_binary_linux_wheel_py3.7_rocm4.3.1
python_version: '3.7' python_version: '3.7'
wheel_docker_image: pytorch/manylinux-rocm:4.2 wheel_docker_image: pytorch/manylinux-rocm:4.3.1
- binary_wheel_upload: - binary_wheel_upload:
context: org-member context: org-member
filters: filters:
...@@ -2117,19 +2117,19 @@ workflows: ...@@ -2117,19 +2117,19 @@ workflows:
only: nightly only: nightly
tags: tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.7_rocm4.2_upload name: nightly_binary_linux_wheel_py3.7_rocm4.3.1_upload
requires: requires:
- nightly_binary_linux_wheel_py3.7_rocm4.2 - nightly_binary_linux_wheel_py3.7_rocm4.3.1
subfolder: rocm4.2/ subfolder: rocm4.3.1/
- smoke_test_linux_pip: - smoke_test_linux_pip:
filters: filters:
branches: branches:
only: only:
- nightly - nightly
name: nightly_binary_linux_wheel_py3.7_rocm4.2_smoke_test_pip name: nightly_binary_linux_wheel_py3.7_rocm4.3.1_smoke_test_pip
python_version: '3.7' python_version: '3.7'
requires: requires:
- nightly_binary_linux_wheel_py3.7_rocm4.2_upload - nightly_binary_linux_wheel_py3.7_rocm4.3.1_upload
- binary_linux_wheel: - binary_linux_wheel:
conda_docker_image: pytorch/conda-builder:cpu conda_docker_image: pytorch/conda-builder:cpu
cu_version: cpu cu_version: cpu
...@@ -2255,15 +2255,15 @@ workflows: ...@@ -2255,15 +2255,15 @@ workflows:
requires: requires:
- nightly_binary_linux_wheel_py3.8_cu113_upload - nightly_binary_linux_wheel_py3.8_cu113_upload
- binary_linux_wheel: - binary_linux_wheel:
cu_version: rocm4.1 cu_version: rocm4.2
filters: filters:
branches: branches:
only: nightly only: nightly
tags: tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.8_rocm4.1 name: nightly_binary_linux_wheel_py3.8_rocm4.2
python_version: '3.8' python_version: '3.8'
wheel_docker_image: pytorch/manylinux-rocm:4.1 wheel_docker_image: pytorch/manylinux-rocm:4.2
- binary_wheel_upload: - binary_wheel_upload:
context: org-member context: org-member
filters: filters:
...@@ -2271,29 +2271,29 @@ workflows: ...@@ -2271,29 +2271,29 @@ workflows:
only: nightly only: nightly
tags: tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.8_rocm4.1_upload name: nightly_binary_linux_wheel_py3.8_rocm4.2_upload
requires: requires:
- nightly_binary_linux_wheel_py3.8_rocm4.1 - nightly_binary_linux_wheel_py3.8_rocm4.2
subfolder: rocm4.1/ subfolder: rocm4.2/
- smoke_test_linux_pip: - smoke_test_linux_pip:
filters: filters:
branches: branches:
only: only:
- nightly - nightly
name: nightly_binary_linux_wheel_py3.8_rocm4.1_smoke_test_pip name: nightly_binary_linux_wheel_py3.8_rocm4.2_smoke_test_pip
python_version: '3.8' python_version: '3.8'
requires: requires:
- nightly_binary_linux_wheel_py3.8_rocm4.1_upload - nightly_binary_linux_wheel_py3.8_rocm4.2_upload
- binary_linux_wheel: - binary_linux_wheel:
cu_version: rocm4.2 cu_version: rocm4.3.1
filters: filters:
branches: branches:
only: nightly only: nightly
tags: tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.8_rocm4.2 name: nightly_binary_linux_wheel_py3.8_rocm4.3.1
python_version: '3.8' python_version: '3.8'
wheel_docker_image: pytorch/manylinux-rocm:4.2 wheel_docker_image: pytorch/manylinux-rocm:4.3.1
- binary_wheel_upload: - binary_wheel_upload:
context: org-member context: org-member
filters: filters:
...@@ -2301,19 +2301,19 @@ workflows: ...@@ -2301,19 +2301,19 @@ workflows:
only: nightly only: nightly
tags: tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.8_rocm4.2_upload name: nightly_binary_linux_wheel_py3.8_rocm4.3.1_upload
requires: requires:
- nightly_binary_linux_wheel_py3.8_rocm4.2 - nightly_binary_linux_wheel_py3.8_rocm4.3.1
subfolder: rocm4.2/ subfolder: rocm4.3.1/
- smoke_test_linux_pip: - smoke_test_linux_pip:
filters: filters:
branches: branches:
only: only:
- nightly - nightly
name: nightly_binary_linux_wheel_py3.8_rocm4.2_smoke_test_pip name: nightly_binary_linux_wheel_py3.8_rocm4.3.1_smoke_test_pip
python_version: '3.8' python_version: '3.8'
requires: requires:
- nightly_binary_linux_wheel_py3.8_rocm4.2_upload - nightly_binary_linux_wheel_py3.8_rocm4.3.1_upload
- binary_linux_wheel: - binary_linux_wheel:
conda_docker_image: pytorch/conda-builder:cpu conda_docker_image: pytorch/conda-builder:cpu
cu_version: cpu cu_version: cpu
...@@ -2439,15 +2439,15 @@ workflows: ...@@ -2439,15 +2439,15 @@ workflows:
requires: requires:
- nightly_binary_linux_wheel_py3.9_cu113_upload - nightly_binary_linux_wheel_py3.9_cu113_upload
- binary_linux_wheel: - binary_linux_wheel:
cu_version: rocm4.1 cu_version: rocm4.2
filters: filters:
branches: branches:
only: nightly only: nightly
tags: tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.9_rocm4.1 name: nightly_binary_linux_wheel_py3.9_rocm4.2
python_version: '3.9' python_version: '3.9'
wheel_docker_image: pytorch/manylinux-rocm:4.1 wheel_docker_image: pytorch/manylinux-rocm:4.2
- binary_wheel_upload: - binary_wheel_upload:
context: org-member context: org-member
filters: filters:
...@@ -2455,29 +2455,29 @@ workflows: ...@@ -2455,29 +2455,29 @@ workflows:
only: nightly only: nightly
tags: tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.9_rocm4.1_upload name: nightly_binary_linux_wheel_py3.9_rocm4.2_upload
requires: requires:
- nightly_binary_linux_wheel_py3.9_rocm4.1 - nightly_binary_linux_wheel_py3.9_rocm4.2
subfolder: rocm4.1/ subfolder: rocm4.2/
- smoke_test_linux_pip: - smoke_test_linux_pip:
filters: filters:
branches: branches:
only: only:
- nightly - nightly
name: nightly_binary_linux_wheel_py3.9_rocm4.1_smoke_test_pip name: nightly_binary_linux_wheel_py3.9_rocm4.2_smoke_test_pip
python_version: '3.9' python_version: '3.9'
requires: requires:
- nightly_binary_linux_wheel_py3.9_rocm4.1_upload - nightly_binary_linux_wheel_py3.9_rocm4.2_upload
- binary_linux_wheel: - binary_linux_wheel:
cu_version: rocm4.2 cu_version: rocm4.3.1
filters: filters:
branches: branches:
only: nightly only: nightly
tags: tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.9_rocm4.2 name: nightly_binary_linux_wheel_py3.9_rocm4.3.1
python_version: '3.9' python_version: '3.9'
wheel_docker_image: pytorch/manylinux-rocm:4.2 wheel_docker_image: pytorch/manylinux-rocm:4.3.1
- binary_wheel_upload: - binary_wheel_upload:
context: org-member context: org-member
filters: filters:
...@@ -2485,19 +2485,19 @@ workflows: ...@@ -2485,19 +2485,19 @@ workflows:
only: nightly only: nightly
tags: tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.9_rocm4.2_upload name: nightly_binary_linux_wheel_py3.9_rocm4.3.1_upload
requires: requires:
- nightly_binary_linux_wheel_py3.9_rocm4.2 - nightly_binary_linux_wheel_py3.9_rocm4.3.1
subfolder: rocm4.2/ subfolder: rocm4.3.1/
- smoke_test_linux_pip: - smoke_test_linux_pip:
filters: filters:
branches: branches:
only: only:
- nightly - nightly
name: nightly_binary_linux_wheel_py3.9_rocm4.2_smoke_test_pip name: nightly_binary_linux_wheel_py3.9_rocm4.3.1_smoke_test_pip
python_version: '3.9' python_version: '3.9'
requires: requires:
- nightly_binary_linux_wheel_py3.9_rocm4.2_upload - nightly_binary_linux_wheel_py3.9_rocm4.3.1_upload
- binary_macos_wheel: - binary_macos_wheel:
conda_docker_image: pytorch/conda-builder:cpu conda_docker_image: pytorch/conda-builder:cpu
cu_version: cpu cu_version: cpu
......
.circleci/regenerate.py
View file @ 87c5e572
...@@ -32,7 +32,7 @@ def build_workflows(prefix="", filter_branch=None, upload=False, indentation=6, ...@@ -32,7 +32,7 @@ def build_workflows(prefix="", filter_branch=None, upload=False, indentation=6,
for os_type in ["linux", "macos", "win"]: for os_type in ["linux", "macos", "win"]:
python_versions = PYTHON_VERSIONS python_versions = PYTHON_VERSIONS
cu_versions_dict = { cu_versions_dict = {
"linux": ["cpu", "cu102", "cu111", "cu113", "rocm4.1", "rocm4.2"], "linux": ["cpu", "cu102", "cu111", "cu113", "rocm4.2", "rocm4.3.1"],
"win": ["cpu", "cu102", "cu111", "cu113"], "win": ["cpu", "cu102", "cu111", "cu113"],
"macos": ["cpu"], "macos": ["cpu"],
} }
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment