Unverified Commit c9493885 authored by Jithun Nair's avatar Jithun Nair Committed by GitHub
Browse files

Update to rocm4.5.2 wheels (#5179)


Co-authored-by: default avatarVasilis Vryniotis <datumbox@users.noreply.github.com>
Co-authored-by: default avatarNikita Shulga <nshulga@fb.com>
parent 729178c7
......@@ -1119,16 +1119,16 @@ workflows:
name: binary_linux_wheel_py3.7_cu115
python_version: '3.7'
wheel_docker_image: pytorch/manylinux-cuda115
- binary_linux_wheel:
cu_version: rocm4.2
name: binary_linux_wheel_py3.7_rocm4.2
python_version: '3.7'
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:
cu_version: rocm4.5.2
name: binary_linux_wheel_py3.7_rocm4.5.2
python_version: '3.7'
wheel_docker_image: pytorch/manylinux-rocm:4.5.2
- binary_linux_wheel:
conda_docker_image: pytorch/conda-builder:cpu
cu_version: cpu
......@@ -1159,16 +1159,16 @@ workflows:
name: binary_linux_wheel_py3.8_cu115
python_version: '3.8'
wheel_docker_image: pytorch/manylinux-cuda115
- binary_linux_wheel:
cu_version: rocm4.2
name: binary_linux_wheel_py3.8_rocm4.2
python_version: '3.8'
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:
cu_version: rocm4.5.2
name: binary_linux_wheel_py3.8_rocm4.5.2
python_version: '3.8'
wheel_docker_image: pytorch/manylinux-rocm:4.5.2
- binary_linux_wheel:
conda_docker_image: pytorch/conda-builder:cpu
cu_version: cpu
......@@ -1199,16 +1199,16 @@ workflows:
name: binary_linux_wheel_py3.9_cu115
python_version: '3.9'
wheel_docker_image: pytorch/manylinux-cuda115
- binary_linux_wheel:
cu_version: rocm4.2
name: binary_linux_wheel_py3.9_rocm4.2
python_version: '3.9'
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_linux_wheel:
cu_version: rocm4.5.2
name: binary_linux_wheel_py3.9_rocm4.5.2
python_version: '3.9'
wheel_docker_image: pytorch/manylinux-rocm:4.5.2
- binary_macos_wheel:
conda_docker_image: pytorch/conda-builder:cpu
cu_version: cpu
......@@ -1872,15 +1872,15 @@ workflows:
requires:
- nightly_binary_linux_wheel_py3.7_cu115_upload
- binary_linux_wheel:
cu_version: rocm4.2
cu_version: rocm4.3.1
filters:
branches:
only: nightly
tags:
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'
wheel_docker_image: pytorch/manylinux-rocm:4.2
wheel_docker_image: pytorch/manylinux-rocm:4.3.1
- binary_wheel_upload:
context: org-member
filters:
......@@ -1888,29 +1888,29 @@ workflows:
only: nightly
tags:
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:
- nightly_binary_linux_wheel_py3.7_rocm4.2
subfolder: rocm4.2/
- nightly_binary_linux_wheel_py3.7_rocm4.3.1
subfolder: rocm4.3.1/
- smoke_test_linux_pip:
filters:
branches:
only:
- 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'
requires:
- nightly_binary_linux_wheel_py3.7_rocm4.2_upload
- nightly_binary_linux_wheel_py3.7_rocm4.3.1_upload
- binary_linux_wheel:
cu_version: rocm4.3.1
cu_version: rocm4.5.2
filters:
branches:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.7_rocm4.3.1
name: nightly_binary_linux_wheel_py3.7_rocm4.5.2
python_version: '3.7'
wheel_docker_image: pytorch/manylinux-rocm:4.3.1
wheel_docker_image: pytorch/manylinux-rocm:4.5.2
- binary_wheel_upload:
context: org-member
filters:
......@@ -1918,19 +1918,19 @@ workflows:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.7_rocm4.3.1_upload
name: nightly_binary_linux_wheel_py3.7_rocm4.5.2_upload
requires:
- nightly_binary_linux_wheel_py3.7_rocm4.3.1
subfolder: rocm4.3.1/
- nightly_binary_linux_wheel_py3.7_rocm4.5.2
subfolder: rocm4.5.2/
- smoke_test_linux_pip:
filters:
branches:
only:
- nightly
name: nightly_binary_linux_wheel_py3.7_rocm4.3.1_smoke_test_pip
name: nightly_binary_linux_wheel_py3.7_rocm4.5.2_smoke_test_pip
python_version: '3.7'
requires:
- nightly_binary_linux_wheel_py3.7_rocm4.3.1_upload
- nightly_binary_linux_wheel_py3.7_rocm4.5.2_upload
- binary_linux_wheel:
conda_docker_image: pytorch/conda-builder:cpu
cu_version: cpu
......@@ -2087,15 +2087,15 @@ workflows:
requires:
- nightly_binary_linux_wheel_py3.8_cu115_upload
- binary_linux_wheel:
cu_version: rocm4.2
cu_version: rocm4.3.1
filters:
branches:
only: nightly
tags:
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'
wheel_docker_image: pytorch/manylinux-rocm:4.2
wheel_docker_image: pytorch/manylinux-rocm:4.3.1
- binary_wheel_upload:
context: org-member
filters:
......@@ -2103,29 +2103,29 @@ workflows:
only: nightly
tags:
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:
- nightly_binary_linux_wheel_py3.8_rocm4.2
subfolder: rocm4.2/
- nightly_binary_linux_wheel_py3.8_rocm4.3.1
subfolder: rocm4.3.1/
- smoke_test_linux_pip:
filters:
branches:
only:
- 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'
requires:
- nightly_binary_linux_wheel_py3.8_rocm4.2_upload
- nightly_binary_linux_wheel_py3.8_rocm4.3.1_upload
- binary_linux_wheel:
cu_version: rocm4.3.1
cu_version: rocm4.5.2
filters:
branches:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.8_rocm4.3.1
name: nightly_binary_linux_wheel_py3.8_rocm4.5.2
python_version: '3.8'
wheel_docker_image: pytorch/manylinux-rocm:4.3.1
wheel_docker_image: pytorch/manylinux-rocm:4.5.2
- binary_wheel_upload:
context: org-member
filters:
......@@ -2133,19 +2133,19 @@ workflows:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.8_rocm4.3.1_upload
name: nightly_binary_linux_wheel_py3.8_rocm4.5.2_upload
requires:
- nightly_binary_linux_wheel_py3.8_rocm4.3.1
subfolder: rocm4.3.1/
- nightly_binary_linux_wheel_py3.8_rocm4.5.2
subfolder: rocm4.5.2/
- smoke_test_linux_pip:
filters:
branches:
only:
- nightly
name: nightly_binary_linux_wheel_py3.8_rocm4.3.1_smoke_test_pip
name: nightly_binary_linux_wheel_py3.8_rocm4.5.2_smoke_test_pip
python_version: '3.8'
requires:
- nightly_binary_linux_wheel_py3.8_rocm4.3.1_upload
- nightly_binary_linux_wheel_py3.8_rocm4.5.2_upload
- binary_linux_wheel:
conda_docker_image: pytorch/conda-builder:cpu
cu_version: cpu
......@@ -2302,15 +2302,15 @@ workflows:
requires:
- nightly_binary_linux_wheel_py3.9_cu115_upload
- binary_linux_wheel:
cu_version: rocm4.2
cu_version: rocm4.3.1
filters:
branches:
only: nightly
tags:
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'
wheel_docker_image: pytorch/manylinux-rocm:4.2
wheel_docker_image: pytorch/manylinux-rocm:4.3.1
- binary_wheel_upload:
context: org-member
filters:
......@@ -2318,29 +2318,29 @@ workflows:
only: nightly
tags:
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:
- nightly_binary_linux_wheel_py3.9_rocm4.2
subfolder: rocm4.2/
- nightly_binary_linux_wheel_py3.9_rocm4.3.1
subfolder: rocm4.3.1/
- smoke_test_linux_pip:
filters:
branches:
only:
- 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'
requires:
- nightly_binary_linux_wheel_py3.9_rocm4.2_upload
- nightly_binary_linux_wheel_py3.9_rocm4.3.1_upload
- binary_linux_wheel:
cu_version: rocm4.3.1
cu_version: rocm4.5.2
filters:
branches:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.9_rocm4.3.1
name: nightly_binary_linux_wheel_py3.9_rocm4.5.2
python_version: '3.9'
wheel_docker_image: pytorch/manylinux-rocm:4.3.1
wheel_docker_image: pytorch/manylinux-rocm:4.5.2
- binary_wheel_upload:
context: org-member
filters:
......@@ -2348,19 +2348,19 @@ workflows:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.9_rocm4.3.1_upload
name: nightly_binary_linux_wheel_py3.9_rocm4.5.2_upload
requires:
- nightly_binary_linux_wheel_py3.9_rocm4.3.1
subfolder: rocm4.3.1/
- nightly_binary_linux_wheel_py3.9_rocm4.5.2
subfolder: rocm4.5.2/
- smoke_test_linux_pip:
filters:
branches:
only:
- nightly
name: nightly_binary_linux_wheel_py3.9_rocm4.3.1_smoke_test_pip
name: nightly_binary_linux_wheel_py3.9_rocm4.5.2_smoke_test_pip
python_version: '3.9'
requires:
- nightly_binary_linux_wheel_py3.9_rocm4.3.1_upload
- nightly_binary_linux_wheel_py3.9_rocm4.5.2_upload
- binary_macos_wheel:
conda_docker_image: pytorch/conda-builder:cpu
cu_version: cpu
......
......@@ -32,7 +32,7 @@ def build_workflows(prefix="", filter_branch=None, upload=False, indentation=6,
for os_type in ["linux", "macos", "win"]:
python_versions = PYTHON_VERSIONS
cu_versions_dict = {
"linux": ["cpu", "cu102", "cu111", "cu113", "cu115", "rocm4.2", "rocm4.3.1"],
"linux": ["cpu", "cu102", "cu111", "cu113", "cu115", "rocm4.3.1", "rocm4.5.2"],
"win": ["cpu", "cu111", "cu113", "cu115"],
"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