Skip to content

GitLab

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

Update to rocm5.2 wheels (#6571) 


Co-authored-by: default avatarVasilis Vryniotis <datumbox@users.noreply.github.com>
parent a5b36d3b
Show whitespace changes
Inline Side-by-side
Showing with 69 additions and 69 deletions
.circleci/config.yml
View file @ b76eb266
......@@ -1126,16 +1126,16 @@ workflows:
name: binary_linux_wheel_py3.7_cu117
python_version: '3.7'
wheel_docker_image: pytorch/manylinux-cuda117
- binary_linux_wheel:
cu_version: rocm5.0
name: binary_linux_wheel_py3.7_rocm5.0
python_version: '3.7'
wheel_docker_image: pytorch/manylinux-rocm:5.0
- binary_linux_wheel:
cu_version: rocm5.1.1
name: binary_linux_wheel_py3.7_rocm5.1.1
python_version: '3.7'
wheel_docker_image: pytorch/manylinux-rocm:5.1.1
- binary_linux_wheel:
cu_version: rocm5.2
name: binary_linux_wheel_py3.7_rocm5.2
python_version: '3.7'
wheel_docker_image: pytorch/manylinux-rocm:5.2
- binary_linux_wheel:
conda_docker_image: pytorch/conda-builder:cpu
cu_version: cpu
......@@ -1166,16 +1166,16 @@ workflows:
name: binary_linux_wheel_py3.8_cu117
python_version: '3.8'
wheel_docker_image: pytorch/manylinux-cuda117
- binary_linux_wheel:
cu_version: rocm5.0
name: binary_linux_wheel_py3.8_rocm5.0
python_version: '3.8'
wheel_docker_image: pytorch/manylinux-rocm:5.0
- binary_linux_wheel:
cu_version: rocm5.1.1
name: binary_linux_wheel_py3.8_rocm5.1.1
python_version: '3.8'
wheel_docker_image: pytorch/manylinux-rocm:5.1.1
- binary_linux_wheel:
cu_version: rocm5.2
name: binary_linux_wheel_py3.8_rocm5.2
python_version: '3.8'
wheel_docker_image: pytorch/manylinux-rocm:5.2
- binary_linux_wheel:
conda_docker_image: pytorch/conda-builder:cpu
cu_version: cpu
......@@ -1206,16 +1206,16 @@ workflows:
name: binary_linux_wheel_py3.9_cu117
python_version: '3.9'
wheel_docker_image: pytorch/manylinux-cuda117
- binary_linux_wheel:
cu_version: rocm5.0
name: binary_linux_wheel_py3.9_rocm5.0
python_version: '3.9'
wheel_docker_image: pytorch/manylinux-rocm:5.0
- binary_linux_wheel:
cu_version: rocm5.1.1
name: binary_linux_wheel_py3.9_rocm5.1.1
python_version: '3.9'
wheel_docker_image: pytorch/manylinux-rocm:5.1.1
- binary_linux_wheel:
cu_version: rocm5.2
name: binary_linux_wheel_py3.9_rocm5.2
python_version: '3.9'
wheel_docker_image: pytorch/manylinux-rocm:5.2
- binary_linux_wheel:
conda_docker_image: pytorch/conda-builder:cpu
cu_version: cpu
......@@ -1246,16 +1246,16 @@ workflows:
name: binary_linux_wheel_py3.10_cu117
python_version: '3.10'
wheel_docker_image: pytorch/manylinux-cuda117
- binary_linux_wheel:
cu_version: rocm5.0
name: binary_linux_wheel_py3.10_rocm5.0
python_version: '3.10'
wheel_docker_image: pytorch/manylinux-rocm:5.0
- binary_linux_wheel:
cu_version: rocm5.1.1
name: binary_linux_wheel_py3.10_rocm5.1.1
python_version: '3.10'
wheel_docker_image: pytorch/manylinux-rocm:5.1.1
- binary_linux_wheel:
cu_version: rocm5.2
name: binary_linux_wheel_py3.10_rocm5.2
python_version: '3.10'
wheel_docker_image: pytorch/manylinux-rocm:5.2
- binary_macos_wheel:
conda_docker_image: pytorch/conda-builder:cpu
cu_version: cpu
......@@ -2018,15 +2018,15 @@ workflows:
- nightly_binary_linux_wheel_py3.7_cu117
subfolder: cu117/
- binary_linux_wheel:
cu_version: rocm5.0
cu_version: rocm5.1.1
filters:
branches:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.7_rocm5.0
name: nightly_binary_linux_wheel_py3.7_rocm5.1.1
python_version: '3.7'
wheel_docker_image: pytorch/manylinux-rocm:5.0
wheel_docker_image: pytorch/manylinux-rocm:5.1.1
- binary_wheel_upload:
context: org-member
filters:
......@@ -2034,20 +2034,20 @@ workflows:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.7_rocm5.0_upload
name: nightly_binary_linux_wheel_py3.7_rocm5.1.1_upload
requires:
- nightly_binary_linux_wheel_py3.7_rocm5.0
subfolder: rocm5.0/
- nightly_binary_linux_wheel_py3.7_rocm5.1.1
subfolder: rocm5.1.1/
- binary_linux_wheel:
cu_version: rocm5.1.1
cu_version: rocm5.2
filters:
branches:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.7_rocm5.1.1
name: nightly_binary_linux_wheel_py3.7_rocm5.2
python_version: '3.7'
wheel_docker_image: pytorch/manylinux-rocm:5.1.1
wheel_docker_image: pytorch/manylinux-rocm:5.2
- binary_wheel_upload:
context: org-member
filters:
......@@ -2055,10 +2055,10 @@ workflows:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.7_rocm5.1.1_upload
name: nightly_binary_linux_wheel_py3.7_rocm5.2_upload
requires:
- nightly_binary_linux_wheel_py3.7_rocm5.1.1
subfolder: rocm5.1.1/
- nightly_binary_linux_wheel_py3.7_rocm5.2
subfolder: rocm5.2/
- binary_linux_wheel:
conda_docker_image: pytorch/conda-builder:cpu
cu_version: cpu
......@@ -2170,15 +2170,15 @@ workflows:
- nightly_binary_linux_wheel_py3.8_cu117
subfolder: cu117/
- binary_linux_wheel:
cu_version: rocm5.0
cu_version: rocm5.1.1
filters:
branches:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.8_rocm5.0
name: nightly_binary_linux_wheel_py3.8_rocm5.1.1
python_version: '3.8'
wheel_docker_image: pytorch/manylinux-rocm:5.0
wheel_docker_image: pytorch/manylinux-rocm:5.1.1
- binary_wheel_upload:
context: org-member
filters:
......@@ -2186,20 +2186,20 @@ workflows:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.8_rocm5.0_upload
name: nightly_binary_linux_wheel_py3.8_rocm5.1.1_upload
requires:
- nightly_binary_linux_wheel_py3.8_rocm5.0
subfolder: rocm5.0/
- nightly_binary_linux_wheel_py3.8_rocm5.1.1
subfolder: rocm5.1.1/
- binary_linux_wheel:
cu_version: rocm5.1.1
cu_version: rocm5.2
filters:
branches:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.8_rocm5.1.1
name: nightly_binary_linux_wheel_py3.8_rocm5.2
python_version: '3.8'
wheel_docker_image: pytorch/manylinux-rocm:5.1.1
wheel_docker_image: pytorch/manylinux-rocm:5.2
- binary_wheel_upload:
context: org-member
filters:
......@@ -2207,10 +2207,10 @@ workflows:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.8_rocm5.1.1_upload
name: nightly_binary_linux_wheel_py3.8_rocm5.2_upload
requires:
- nightly_binary_linux_wheel_py3.8_rocm5.1.1
subfolder: rocm5.1.1/
- nightly_binary_linux_wheel_py3.8_rocm5.2
subfolder: rocm5.2/
- binary_linux_wheel:
conda_docker_image: pytorch/conda-builder:cpu
cu_version: cpu
......@@ -2322,15 +2322,15 @@ workflows:
- nightly_binary_linux_wheel_py3.9_cu117
subfolder: cu117/
- binary_linux_wheel:
cu_version: rocm5.0
cu_version: rocm5.1.1
filters:
branches:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.9_rocm5.0
name: nightly_binary_linux_wheel_py3.9_rocm5.1.1
python_version: '3.9'
wheel_docker_image: pytorch/manylinux-rocm:5.0
wheel_docker_image: pytorch/manylinux-rocm:5.1.1
- binary_wheel_upload:
context: org-member
filters:
......@@ -2338,20 +2338,20 @@ workflows:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.9_rocm5.0_upload
name: nightly_binary_linux_wheel_py3.9_rocm5.1.1_upload
requires:
- nightly_binary_linux_wheel_py3.9_rocm5.0
subfolder: rocm5.0/
- nightly_binary_linux_wheel_py3.9_rocm5.1.1
subfolder: rocm5.1.1/
- binary_linux_wheel:
cu_version: rocm5.1.1
cu_version: rocm5.2
filters:
branches:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.9_rocm5.1.1
name: nightly_binary_linux_wheel_py3.9_rocm5.2
python_version: '3.9'
wheel_docker_image: pytorch/manylinux-rocm:5.1.1
wheel_docker_image: pytorch/manylinux-rocm:5.2
- binary_wheel_upload:
context: org-member
filters:
......@@ -2359,10 +2359,10 @@ workflows:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.9_rocm5.1.1_upload
name: nightly_binary_linux_wheel_py3.9_rocm5.2_upload
requires:
- nightly_binary_linux_wheel_py3.9_rocm5.1.1
subfolder: rocm5.1.1/
- nightly_binary_linux_wheel_py3.9_rocm5.2
subfolder: rocm5.2/
- binary_linux_wheel:
conda_docker_image: pytorch/conda-builder:cpu
cu_version: cpu
......@@ -2474,15 +2474,15 @@ workflows:
- nightly_binary_linux_wheel_py3.10_cu117
subfolder: cu117/
- binary_linux_wheel:
cu_version: rocm5.0
cu_version: rocm5.1.1
filters:
branches:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.10_rocm5.0
name: nightly_binary_linux_wheel_py3.10_rocm5.1.1
python_version: '3.10'
wheel_docker_image: pytorch/manylinux-rocm:5.0
wheel_docker_image: pytorch/manylinux-rocm:5.1.1
- binary_wheel_upload:
context: org-member
filters:
......@@ -2490,20 +2490,20 @@ workflows:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.10_rocm5.0_upload
name: nightly_binary_linux_wheel_py3.10_rocm5.1.1_upload
requires:
- nightly_binary_linux_wheel_py3.10_rocm5.0
subfolder: rocm5.0/
- nightly_binary_linux_wheel_py3.10_rocm5.1.1
subfolder: rocm5.1.1/
- binary_linux_wheel:
cu_version: rocm5.1.1
cu_version: rocm5.2
filters:
branches:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.10_rocm5.1.1
name: nightly_binary_linux_wheel_py3.10_rocm5.2
python_version: '3.10'
wheel_docker_image: pytorch/manylinux-rocm:5.1.1
wheel_docker_image: pytorch/manylinux-rocm:5.2
- binary_wheel_upload:
context: org-member
filters:
......@@ -2511,10 +2511,10 @@ workflows:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.10_rocm5.1.1_upload
name: nightly_binary_linux_wheel_py3.10_rocm5.2_upload
requires:
- nightly_binary_linux_wheel_py3.10_rocm5.1.1
subfolder: rocm5.1.1/
- nightly_binary_linux_wheel_py3.10_rocm5.2
subfolder: rocm5.2/
- binary_macos_wheel:
conda_docker_image: pytorch/conda-builder:cpu
cu_version: cpu
......
.circleci/regenerate.py
View file @ b76eb266
......@@ -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", "cu113", "cu116", "cu117", "rocm5.0", "rocm5.1.1"],
"linux": ["cpu", "cu102", "cu113", "cu116", "cu117", "rocm5.1.1", "rocm5.2"],
"win": ["cpu", "cu113", "cu116", "cu117"],
"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