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
Hide whitespace changes
Inline Side-by-side
Showing with 69 additions and 69 deletions
.circleci/config.yml
View file @ b76eb266
...@@ -1126,16 +1126,16 @@ workflows: ...@@ -1126,16 +1126,16 @@ workflows:
name: binary_linux_wheel_py3.7_cu117 name: binary_linux_wheel_py3.7_cu117
python_version: '3.7' python_version: '3.7'
wheel_docker_image: pytorch/manylinux-cuda117 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: - binary_linux_wheel:
cu_version: rocm5.1.1 cu_version: rocm5.1.1
name: binary_linux_wheel_py3.7_rocm5.1.1 name: binary_linux_wheel_py3.7_rocm5.1.1
python_version: '3.7' python_version: '3.7'
wheel_docker_image: pytorch/manylinux-rocm:5.1.1 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: - binary_linux_wheel:
conda_docker_image: pytorch/conda-builder:cpu conda_docker_image: pytorch/conda-builder:cpu
cu_version: cpu cu_version: cpu
...@@ -1166,16 +1166,16 @@ workflows: ...@@ -1166,16 +1166,16 @@ workflows:
name: binary_linux_wheel_py3.8_cu117 name: binary_linux_wheel_py3.8_cu117
python_version: '3.8' python_version: '3.8'
wheel_docker_image: pytorch/manylinux-cuda117 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: - binary_linux_wheel:
cu_version: rocm5.1.1 cu_version: rocm5.1.1
name: binary_linux_wheel_py3.8_rocm5.1.1 name: binary_linux_wheel_py3.8_rocm5.1.1
python_version: '3.8' python_version: '3.8'
wheel_docker_image: pytorch/manylinux-rocm:5.1.1 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: - binary_linux_wheel:
conda_docker_image: pytorch/conda-builder:cpu conda_docker_image: pytorch/conda-builder:cpu
cu_version: cpu cu_version: cpu
...@@ -1206,16 +1206,16 @@ workflows: ...@@ -1206,16 +1206,16 @@ workflows:
name: binary_linux_wheel_py3.9_cu117 name: binary_linux_wheel_py3.9_cu117
python_version: '3.9' python_version: '3.9'
wheel_docker_image: pytorch/manylinux-cuda117 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: - binary_linux_wheel:
cu_version: rocm5.1.1 cu_version: rocm5.1.1
name: binary_linux_wheel_py3.9_rocm5.1.1 name: binary_linux_wheel_py3.9_rocm5.1.1
python_version: '3.9' python_version: '3.9'
wheel_docker_image: pytorch/manylinux-rocm:5.1.1 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: - binary_linux_wheel:
conda_docker_image: pytorch/conda-builder:cpu conda_docker_image: pytorch/conda-builder:cpu
cu_version: cpu cu_version: cpu
...@@ -1246,16 +1246,16 @@ workflows: ...@@ -1246,16 +1246,16 @@ workflows:
name: binary_linux_wheel_py3.10_cu117 name: binary_linux_wheel_py3.10_cu117
python_version: '3.10' python_version: '3.10'
wheel_docker_image: pytorch/manylinux-cuda117 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: - binary_linux_wheel:
cu_version: rocm5.1.1 cu_version: rocm5.1.1
name: binary_linux_wheel_py3.10_rocm5.1.1 name: binary_linux_wheel_py3.10_rocm5.1.1
python_version: '3.10' python_version: '3.10'
wheel_docker_image: pytorch/manylinux-rocm:5.1.1 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: - binary_macos_wheel:
conda_docker_image: pytorch/conda-builder:cpu conda_docker_image: pytorch/conda-builder:cpu
cu_version: cpu cu_version: cpu
...@@ -2018,15 +2018,15 @@ workflows: ...@@ -2018,15 +2018,15 @@ workflows:
- nightly_binary_linux_wheel_py3.7_cu117 - nightly_binary_linux_wheel_py3.7_cu117
subfolder: cu117/ subfolder: cu117/
- binary_linux_wheel: - binary_linux_wheel:
cu_version: rocm5.0 cu_version: rocm5.1.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_rocm5.0 name: nightly_binary_linux_wheel_py3.7_rocm5.1.1
python_version: '3.7' python_version: '3.7'
wheel_docker_image: pytorch/manylinux-rocm:5.0 wheel_docker_image: pytorch/manylinux-rocm:5.1.1
- binary_wheel_upload: - binary_wheel_upload:
context: org-member context: org-member
filters: filters:
...@@ -2034,20 +2034,20 @@ workflows: ...@@ -2034,20 +2034,20 @@ 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_rocm5.0_upload name: nightly_binary_linux_wheel_py3.7_rocm5.1.1_upload
requires: requires:
- nightly_binary_linux_wheel_py3.7_rocm5.0 - nightly_binary_linux_wheel_py3.7_rocm5.1.1
subfolder: rocm5.0/ subfolder: rocm5.1.1/
- binary_linux_wheel: - binary_linux_wheel:
cu_version: rocm5.1.1 cu_version: rocm5.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_rocm5.1.1 name: nightly_binary_linux_wheel_py3.7_rocm5.2
python_version: '3.7' python_version: '3.7'
wheel_docker_image: pytorch/manylinux-rocm:5.1.1 wheel_docker_image: pytorch/manylinux-rocm:5.2
- binary_wheel_upload: - binary_wheel_upload:
context: org-member context: org-member
filters: filters:
...@@ -2055,10 +2055,10 @@ workflows: ...@@ -2055,10 +2055,10 @@ 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_rocm5.1.1_upload name: nightly_binary_linux_wheel_py3.7_rocm5.2_upload
requires: requires:
- nightly_binary_linux_wheel_py3.7_rocm5.1.1 - nightly_binary_linux_wheel_py3.7_rocm5.2
subfolder: rocm5.1.1/ subfolder: rocm5.2/
- 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
...@@ -2170,15 +2170,15 @@ workflows: ...@@ -2170,15 +2170,15 @@ workflows:
- nightly_binary_linux_wheel_py3.8_cu117 - nightly_binary_linux_wheel_py3.8_cu117
subfolder: cu117/ subfolder: cu117/
- binary_linux_wheel: - binary_linux_wheel:
cu_version: rocm5.0 cu_version: rocm5.1.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_rocm5.0 name: nightly_binary_linux_wheel_py3.8_rocm5.1.1
python_version: '3.8' python_version: '3.8'
wheel_docker_image: pytorch/manylinux-rocm:5.0 wheel_docker_image: pytorch/manylinux-rocm:5.1.1
- binary_wheel_upload: - binary_wheel_upload:
context: org-member context: org-member
filters: filters:
...@@ -2186,20 +2186,20 @@ workflows: ...@@ -2186,20 +2186,20 @@ 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_rocm5.0_upload name: nightly_binary_linux_wheel_py3.8_rocm5.1.1_upload
requires: requires:
- nightly_binary_linux_wheel_py3.8_rocm5.0 - nightly_binary_linux_wheel_py3.8_rocm5.1.1
subfolder: rocm5.0/ subfolder: rocm5.1.1/
- binary_linux_wheel: - binary_linux_wheel:
cu_version: rocm5.1.1 cu_version: rocm5.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_rocm5.1.1 name: nightly_binary_linux_wheel_py3.8_rocm5.2
python_version: '3.8' python_version: '3.8'
wheel_docker_image: pytorch/manylinux-rocm:5.1.1 wheel_docker_image: pytorch/manylinux-rocm:5.2
- binary_wheel_upload: - binary_wheel_upload:
context: org-member context: org-member
filters: filters:
...@@ -2207,10 +2207,10 @@ workflows: ...@@ -2207,10 +2207,10 @@ 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_rocm5.1.1_upload name: nightly_binary_linux_wheel_py3.8_rocm5.2_upload
requires: requires:
- nightly_binary_linux_wheel_py3.8_rocm5.1.1 - nightly_binary_linux_wheel_py3.8_rocm5.2
subfolder: rocm5.1.1/ subfolder: rocm5.2/
- 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
...@@ -2322,15 +2322,15 @@ workflows: ...@@ -2322,15 +2322,15 @@ workflows:
- nightly_binary_linux_wheel_py3.9_cu117 - nightly_binary_linux_wheel_py3.9_cu117
subfolder: cu117/ subfolder: cu117/
- binary_linux_wheel: - binary_linux_wheel:
cu_version: rocm5.0 cu_version: rocm5.1.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_rocm5.0 name: nightly_binary_linux_wheel_py3.9_rocm5.1.1
python_version: '3.9' python_version: '3.9'
wheel_docker_image: pytorch/manylinux-rocm:5.0 wheel_docker_image: pytorch/manylinux-rocm:5.1.1
- binary_wheel_upload: - binary_wheel_upload:
context: org-member context: org-member
filters: filters:
...@@ -2338,20 +2338,20 @@ workflows: ...@@ -2338,20 +2338,20 @@ 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_rocm5.0_upload name: nightly_binary_linux_wheel_py3.9_rocm5.1.1_upload
requires: requires:
- nightly_binary_linux_wheel_py3.9_rocm5.0 - nightly_binary_linux_wheel_py3.9_rocm5.1.1
subfolder: rocm5.0/ subfolder: rocm5.1.1/
- binary_linux_wheel: - binary_linux_wheel:
cu_version: rocm5.1.1 cu_version: rocm5.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_rocm5.1.1 name: nightly_binary_linux_wheel_py3.9_rocm5.2
python_version: '3.9' python_version: '3.9'
wheel_docker_image: pytorch/manylinux-rocm:5.1.1 wheel_docker_image: pytorch/manylinux-rocm:5.2
- binary_wheel_upload: - binary_wheel_upload:
context: org-member context: org-member
filters: filters:
...@@ -2359,10 +2359,10 @@ workflows: ...@@ -2359,10 +2359,10 @@ 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_rocm5.1.1_upload name: nightly_binary_linux_wheel_py3.9_rocm5.2_upload
requires: requires:
- nightly_binary_linux_wheel_py3.9_rocm5.1.1 - nightly_binary_linux_wheel_py3.9_rocm5.2
subfolder: rocm5.1.1/ subfolder: rocm5.2/
- 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
...@@ -2474,15 +2474,15 @@ workflows: ...@@ -2474,15 +2474,15 @@ workflows:
- nightly_binary_linux_wheel_py3.10_cu117 - nightly_binary_linux_wheel_py3.10_cu117
subfolder: cu117/ subfolder: cu117/
- binary_linux_wheel: - binary_linux_wheel:
cu_version: rocm5.0 cu_version: rocm5.1.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.10_rocm5.0 name: nightly_binary_linux_wheel_py3.10_rocm5.1.1
python_version: '3.10' python_version: '3.10'
wheel_docker_image: pytorch/manylinux-rocm:5.0 wheel_docker_image: pytorch/manylinux-rocm:5.1.1
- binary_wheel_upload: - binary_wheel_upload:
context: org-member context: org-member
filters: filters:
...@@ -2490,20 +2490,20 @@ workflows: ...@@ -2490,20 +2490,20 @@ 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.10_rocm5.0_upload name: nightly_binary_linux_wheel_py3.10_rocm5.1.1_upload
requires: requires:
- nightly_binary_linux_wheel_py3.10_rocm5.0 - nightly_binary_linux_wheel_py3.10_rocm5.1.1
subfolder: rocm5.0/ subfolder: rocm5.1.1/
- binary_linux_wheel: - binary_linux_wheel:
cu_version: rocm5.1.1 cu_version: rocm5.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.10_rocm5.1.1 name: nightly_binary_linux_wheel_py3.10_rocm5.2
python_version: '3.10' python_version: '3.10'
wheel_docker_image: pytorch/manylinux-rocm:5.1.1 wheel_docker_image: pytorch/manylinux-rocm:5.2
- binary_wheel_upload: - binary_wheel_upload:
context: org-member context: org-member
filters: filters:
...@@ -2511,10 +2511,10 @@ workflows: ...@@ -2511,10 +2511,10 @@ 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.10_rocm5.1.1_upload name: nightly_binary_linux_wheel_py3.10_rocm5.2_upload
requires: requires:
- nightly_binary_linux_wheel_py3.10_rocm5.1.1 - nightly_binary_linux_wheel_py3.10_rocm5.2
subfolder: rocm5.1.1/ subfolder: rocm5.2/
- 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 @ b76eb266
...@@ -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", "cu113", "cu116", "cu117", "rocm5.0", "rocm5.1.1"], "linux": ["cpu", "cu102", "cu113", "cu116", "cu117", "rocm5.1.1", "rocm5.2"],
"win": ["cpu", "cu113", "cu116", "cu117"], "win": ["cpu", "cu113", "cu116", "cu117"],
"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