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

Upgrade nightly wheels to ROCm5.3 (#6955)

* Update to ROCm 5.3

* Regenerate config.yml
parent 01c11a05
...@@ -1120,16 +1120,16 @@ workflows: ...@@ -1120,16 +1120,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.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: - binary_linux_wheel:
cu_version: rocm5.2 cu_version: rocm5.2
name: binary_linux_wheel_py3.7_rocm5.2 name: binary_linux_wheel_py3.7_rocm5.2
python_version: '3.7' python_version: '3.7'
wheel_docker_image: pytorch/manylinux-rocm:5.2 wheel_docker_image: pytorch/manylinux-rocm:5.2
- binary_linux_wheel:
cu_version: rocm5.3
name: binary_linux_wheel_py3.7_rocm5.3
python_version: '3.7'
wheel_docker_image: pytorch/manylinux-rocm:5.3
- binary_win_wheel: - binary_win_wheel:
cu_version: cpu cu_version: cpu
filters: filters:
...@@ -1655,25 +1655,25 @@ workflows: ...@@ -1655,25 +1655,25 @@ workflows:
python_version: '3.7' python_version: '3.7'
wheel_docker_image: pytorch/manylinux-cuda117 wheel_docker_image: pytorch/manylinux-cuda117
- 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_linux_wheel: - binary_linux_wheel:
cu_version: rocm5.2 cu_version: rocm5.3
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.2 name: nightly_binary_linux_wheel_py3.7_rocm5.3
python_version: '3.7' python_version: '3.7'
wheel_docker_image: pytorch/manylinux-rocm:5.2 wheel_docker_image: pytorch/manylinux-rocm:5.3
- binary_win_wheel: - binary_win_wheel:
cu_version: cpu cu_version: cpu
filters: filters:
......
...@@ -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", "cu116", "cu117", "rocm5.1.1", "rocm5.2"], "linux": ["cpu", "cu116", "cu117", "rocm5.2", "rocm5.3"],
"win": ["cpu", "cu116", "cu117"], "win": ["cpu", "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