Skip to content

GitLab

"vscode:/vscode.git/clone" did not exist on "d559ad87f379291d153e67e628e9ce183fdcd14c"
Unverified Commit 890450a4 authored by KyleCZH's avatar KyleCZH Committed by GitHub
Browse files

[ROCm] Update to rocm5.0 wheels (#5660) 


Signed-off-by: default avatarKyle Chen <kylechen@amd.com>
Co-authored-by: default avatarVasilis Vryniotis <datumbox@users.noreply.github.com>
parent 31e503f1
Hide whitespace changes
Inline Side-by-side
Showing with 69 additions and 69 deletions
.circleci/config.yml
View file @ 890450a4
......@@ -1141,16 +1141,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.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:
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:
conda_docker_image: pytorch/conda-builder:cpu
cu_version: cpu
......@@ -1175,16 +1175,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.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:
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:
conda_docker_image: pytorch/conda-builder:cpu
cu_version: cpu
......@@ -1209,16 +1209,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.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_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:
conda_docker_image: pytorch/conda-builder:cpu
cu_version: cpu
......@@ -1243,16 +1243,16 @@ workflows:
name: binary_linux_wheel_py3.10_cu115
python_version: '3.10'
wheel_docker_image: pytorch/manylinux-cuda115
- binary_linux_wheel:
cu_version: rocm4.3.1
name: binary_linux_wheel_py3.10_rocm4.3.1
python_version: '3.10'
wheel_docker_image: pytorch/manylinux-rocm:4.3.1
- binary_linux_wheel:
cu_version: rocm4.5.2
name: binary_linux_wheel_py3.10_rocm4.5.2
python_version: '3.10'
wheel_docker_image: pytorch/manylinux-rocm:4.5.2
- 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_macos_wheel:
conda_docker_image: pytorch/conda-builder:cpu
cu_version: cpu
......@@ -1898,15 +1898,15 @@ workflows:
- nightly_binary_linux_wheel_py3.7_cu115
subfolder: cu115/
- 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:
......@@ -1914,20 +1914,20 @@ 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/
- binary_linux_wheel:
cu_version: rocm4.5.2
cu_version: rocm5.0
filters:
branches:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.7_rocm4.5.2
name: nightly_binary_linux_wheel_py3.7_rocm5.0
python_version: '3.7'
wheel_docker_image: pytorch/manylinux-rocm:4.5.2
wheel_docker_image: pytorch/manylinux-rocm:5.0
- binary_wheel_upload:
context: org-member
filters:
......@@ -1935,10 +1935,10 @@ workflows:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.7_rocm4.5.2_upload
name: nightly_binary_linux_wheel_py3.7_rocm5.0_upload
requires:
- nightly_binary_linux_wheel_py3.7_rocm4.5.2
subfolder: rocm4.5.2/
- nightly_binary_linux_wheel_py3.7_rocm5.0
subfolder: rocm5.0/
- binary_linux_wheel:
conda_docker_image: pytorch/conda-builder:cpu
cu_version: cpu
......@@ -2028,15 +2028,15 @@ workflows:
- nightly_binary_linux_wheel_py3.8_cu115
subfolder: cu115/
- 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:
......@@ -2044,20 +2044,20 @@ 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/
- binary_linux_wheel:
cu_version: rocm4.5.2
cu_version: rocm5.0
filters:
branches:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.8_rocm4.5.2
name: nightly_binary_linux_wheel_py3.8_rocm5.0
python_version: '3.8'
wheel_docker_image: pytorch/manylinux-rocm:4.5.2
wheel_docker_image: pytorch/manylinux-rocm:5.0
- binary_wheel_upload:
context: org-member
filters:
......@@ -2065,10 +2065,10 @@ workflows:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.8_rocm4.5.2_upload
name: nightly_binary_linux_wheel_py3.8_rocm5.0_upload
requires:
- nightly_binary_linux_wheel_py3.8_rocm4.5.2
subfolder: rocm4.5.2/
- nightly_binary_linux_wheel_py3.8_rocm5.0
subfolder: rocm5.0/
- binary_linux_wheel:
conda_docker_image: pytorch/conda-builder:cpu
cu_version: cpu
......@@ -2158,15 +2158,15 @@ workflows:
- nightly_binary_linux_wheel_py3.9_cu115
subfolder: cu115/
- 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:
......@@ -2174,20 +2174,20 @@ 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/
- binary_linux_wheel:
cu_version: rocm4.5.2
cu_version: rocm5.0
filters:
branches:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.9_rocm4.5.2
name: nightly_binary_linux_wheel_py3.9_rocm5.0
python_version: '3.9'
wheel_docker_image: pytorch/manylinux-rocm:4.5.2
wheel_docker_image: pytorch/manylinux-rocm:5.0
- binary_wheel_upload:
context: org-member
filters:
......@@ -2195,10 +2195,10 @@ workflows:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.9_rocm4.5.2_upload
name: nightly_binary_linux_wheel_py3.9_rocm5.0_upload
requires:
- nightly_binary_linux_wheel_py3.9_rocm4.5.2
subfolder: rocm4.5.2/
- nightly_binary_linux_wheel_py3.9_rocm5.0
subfolder: rocm5.0/
- binary_linux_wheel:
conda_docker_image: pytorch/conda-builder:cpu
cu_version: cpu
......@@ -2288,15 +2288,15 @@ workflows:
- nightly_binary_linux_wheel_py3.10_cu115
subfolder: cu115/
- 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.10_rocm4.3.1
name: nightly_binary_linux_wheel_py3.10_rocm4.5.2
python_version: '3.10'
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:
......@@ -2304,20 +2304,20 @@ workflows:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.10_rocm4.3.1_upload
name: nightly_binary_linux_wheel_py3.10_rocm4.5.2_upload
requires:
- nightly_binary_linux_wheel_py3.10_rocm4.3.1
subfolder: rocm4.3.1/
- nightly_binary_linux_wheel_py3.10_rocm4.5.2
subfolder: rocm4.5.2/
- binary_linux_wheel:
cu_version: rocm4.5.2
cu_version: rocm5.0
filters:
branches:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.10_rocm4.5.2
name: nightly_binary_linux_wheel_py3.10_rocm5.0
python_version: '3.10'
wheel_docker_image: pytorch/manylinux-rocm:4.5.2
wheel_docker_image: pytorch/manylinux-rocm:5.0
- binary_wheel_upload:
context: org-member
filters:
......@@ -2325,10 +2325,10 @@ workflows:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.10_rocm4.5.2_upload
name: nightly_binary_linux_wheel_py3.10_rocm5.0_upload
requires:
- nightly_binary_linux_wheel_py3.10_rocm4.5.2
subfolder: rocm4.5.2/
- nightly_binary_linux_wheel_py3.10_rocm5.0
subfolder: rocm5.0/
- binary_macos_wheel:
conda_docker_image: pytorch/conda-builder:cpu
cu_version: cpu
......
.circleci/regenerate.py
View file @ 890450a4
......@@ -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", "cu115", "rocm4.3.1", "rocm4.5.2"],
"linux": ["cpu", "cu102", "cu113", "cu115", "rocm4.5.2", "rocm5.0"],
"win": ["cpu", "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