Commit 3f2ffc94 authored by Jithun Nair's avatar Jithun Nair Committed by Facebook GitHub Bot
Browse files

Update nightly wheels to ROCm5.2 (#2672)

Summary: Pull Request resolved: https://github.com/pytorch/audio/pull/2672

Reviewed By: atalman

Differential Revision: D39468320

Pulled By: mthrok

fbshipit-source-id: 0e7bd4fd922ba0db51700e140b95328a5b687a6f
parent ad32eae8
......@@ -969,31 +969,31 @@ workflows:
requires:
- binary_linux_wheel_py3.7_cu117
- binary_linux_wheel:
cuda_version: rocm5.0
name: binary_linux_wheel_py3.7_rocm5.0
cuda_version: rocm5.1.1
name: binary_linux_wheel_py3.7_rocm5.1.1
python_version: '3.7'
requires:
- build_ffmpeg_linux
wheel_docker_image: pytorch/manylinux-rocm:5.0
wheel_docker_image: pytorch/manylinux-rocm:5.1.1
- smoke_test_linux_pip:
cuda_version: rocm5.0
name: binary_linux_wheel_py3.7_rocm5.0_smoke_test_pip
cuda_version: rocm5.1.1
name: binary_linux_wheel_py3.7_rocm5.1.1_smoke_test_pip
python_version: '3.7'
requires:
- binary_linux_wheel_py3.7_rocm5.0
- binary_linux_wheel_py3.7_rocm5.1.1
- binary_linux_wheel:
cuda_version: rocm5.1.1
name: binary_linux_wheel_py3.7_rocm5.1.1
cuda_version: rocm5.2
name: binary_linux_wheel_py3.7_rocm5.2
python_version: '3.7'
requires:
- build_ffmpeg_linux
wheel_docker_image: pytorch/manylinux-rocm:5.1.1
wheel_docker_image: pytorch/manylinux-rocm:5.2
- smoke_test_linux_pip:
cuda_version: rocm5.1.1
name: binary_linux_wheel_py3.7_rocm5.1.1_smoke_test_pip
cuda_version: rocm5.2
name: binary_linux_wheel_py3.7_rocm5.2_smoke_test_pip
python_version: '3.7'
requires:
- binary_linux_wheel_py3.7_rocm5.1.1
- binary_linux_wheel_py3.7_rocm5.2
- binary_linux_wheel:
cuda_version: cpu
filters:
......@@ -1058,31 +1058,31 @@ workflows:
requires:
- binary_linux_wheel_py3.8_cu117
- binary_linux_wheel:
cuda_version: rocm5.0
name: binary_linux_wheel_py3.8_rocm5.0
cuda_version: rocm5.1.1
name: binary_linux_wheel_py3.8_rocm5.1.1
python_version: '3.8'
requires:
- build_ffmpeg_linux
wheel_docker_image: pytorch/manylinux-rocm:5.0
wheel_docker_image: pytorch/manylinux-rocm:5.1.1
- smoke_test_linux_pip:
cuda_version: rocm5.0
name: binary_linux_wheel_py3.8_rocm5.0_smoke_test_pip
cuda_version: rocm5.1.1
name: binary_linux_wheel_py3.8_rocm5.1.1_smoke_test_pip
python_version: '3.8'
requires:
- binary_linux_wheel_py3.8_rocm5.0
- binary_linux_wheel_py3.8_rocm5.1.1
- binary_linux_wheel:
cuda_version: rocm5.1.1
name: binary_linux_wheel_py3.8_rocm5.1.1
cuda_version: rocm5.2
name: binary_linux_wheel_py3.8_rocm5.2
python_version: '3.8'
requires:
- build_ffmpeg_linux
wheel_docker_image: pytorch/manylinux-rocm:5.1.1
wheel_docker_image: pytorch/manylinux-rocm:5.2
- smoke_test_linux_pip:
cuda_version: rocm5.1.1
name: binary_linux_wheel_py3.8_rocm5.1.1_smoke_test_pip
cuda_version: rocm5.2
name: binary_linux_wheel_py3.8_rocm5.2_smoke_test_pip
python_version: '3.8'
requires:
- binary_linux_wheel_py3.8_rocm5.1.1
- binary_linux_wheel_py3.8_rocm5.2
- binary_linux_wheel:
cuda_version: cpu
name: binary_linux_wheel_py3.9_cpu
......@@ -1135,31 +1135,31 @@ workflows:
requires:
- binary_linux_wheel_py3.9_cu117
- binary_linux_wheel:
cuda_version: rocm5.0
name: binary_linux_wheel_py3.9_rocm5.0
cuda_version: rocm5.1.1
name: binary_linux_wheel_py3.9_rocm5.1.1
python_version: '3.9'
requires:
- build_ffmpeg_linux
wheel_docker_image: pytorch/manylinux-rocm:5.0
wheel_docker_image: pytorch/manylinux-rocm:5.1.1
- smoke_test_linux_pip:
cuda_version: rocm5.0
name: binary_linux_wheel_py3.9_rocm5.0_smoke_test_pip
cuda_version: rocm5.1.1
name: binary_linux_wheel_py3.9_rocm5.1.1_smoke_test_pip
python_version: '3.9'
requires:
- binary_linux_wheel_py3.9_rocm5.0
- binary_linux_wheel_py3.9_rocm5.1.1
- binary_linux_wheel:
cuda_version: rocm5.1.1
name: binary_linux_wheel_py3.9_rocm5.1.1
cuda_version: rocm5.2
name: binary_linux_wheel_py3.9_rocm5.2
python_version: '3.9'
requires:
- build_ffmpeg_linux
wheel_docker_image: pytorch/manylinux-rocm:5.1.1
wheel_docker_image: pytorch/manylinux-rocm:5.2
- smoke_test_linux_pip:
cuda_version: rocm5.1.1
name: binary_linux_wheel_py3.9_rocm5.1.1_smoke_test_pip
cuda_version: rocm5.2
name: binary_linux_wheel_py3.9_rocm5.2_smoke_test_pip
python_version: '3.9'
requires:
- binary_linux_wheel_py3.9_rocm5.1.1
- binary_linux_wheel_py3.9_rocm5.2
- binary_linux_wheel:
cuda_version: cpu
name: binary_linux_wheel_py3.10_cpu
......@@ -1212,31 +1212,31 @@ workflows:
requires:
- binary_linux_wheel_py3.10_cu117
- binary_linux_wheel:
cuda_version: rocm5.0
name: binary_linux_wheel_py3.10_rocm5.0
cuda_version: rocm5.1.1
name: binary_linux_wheel_py3.10_rocm5.1.1
python_version: '3.10'
requires:
- build_ffmpeg_linux
wheel_docker_image: pytorch/manylinux-rocm:5.0
wheel_docker_image: pytorch/manylinux-rocm:5.1.1
- smoke_test_linux_pip:
cuda_version: rocm5.0
name: binary_linux_wheel_py3.10_rocm5.0_smoke_test_pip
cuda_version: rocm5.1.1
name: binary_linux_wheel_py3.10_rocm5.1.1_smoke_test_pip
python_version: '3.10'
requires:
- binary_linux_wheel_py3.10_rocm5.0
- binary_linux_wheel_py3.10_rocm5.1.1
- binary_linux_wheel:
cuda_version: rocm5.1.1
name: binary_linux_wheel_py3.10_rocm5.1.1
cuda_version: rocm5.2
name: binary_linux_wheel_py3.10_rocm5.2
python_version: '3.10'
requires:
- build_ffmpeg_linux
wheel_docker_image: pytorch/manylinux-rocm:5.1.1
wheel_docker_image: pytorch/manylinux-rocm:5.2
- smoke_test_linux_pip:
cuda_version: rocm5.1.1
name: binary_linux_wheel_py3.10_rocm5.1.1_smoke_test_pip
cuda_version: rocm5.2
name: binary_linux_wheel_py3.10_rocm5.2_smoke_test_pip
python_version: '3.10'
requires:
- binary_linux_wheel_py3.10_rocm5.1.1
- binary_linux_wheel_py3.10_rocm5.2
- binary_macos_wheel:
cuda_version: cpu
name: binary_macos_wheel_py3.7_cpu
......@@ -2103,18 +2103,18 @@ workflows:
requires:
- nightly_binary_linux_wheel_py3.7_cu117
- binary_linux_wheel:
cuda_version: rocm5.0
cuda_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'
requires:
- build_ffmpeg_linux
wheel_docker_image: pytorch/manylinux-rocm:5.0
wheel_docker_image: pytorch/manylinux-rocm:5.1.1
- binary_wheel_upload:
context: org-member
filters:
......@@ -2123,35 +2123,35 @@ workflows:
- 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/
- smoke_test_linux_pip:
cuda_version: rocm5.0
cuda_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_smoke_test_pip
name: nightly_binary_linux_wheel_py3.7_rocm5.1.1_smoke_test_pip
python_version: '3.7'
requires:
- nightly_binary_linux_wheel_py3.7_rocm5.0
- nightly_binary_linux_wheel_py3.7_rocm5.1.1
- binary_linux_wheel:
cuda_version: rocm5.1.1
cuda_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'
requires:
- build_ffmpeg_linux
wheel_docker_image: pytorch/manylinux-rocm:5.1.1
wheel_docker_image: pytorch/manylinux-rocm:5.2
- binary_wheel_upload:
context: org-member
filters:
......@@ -2160,22 +2160,22 @@ workflows:
- 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/
- smoke_test_linux_pip:
cuda_version: rocm5.1.1
cuda_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_smoke_test_pip
name: nightly_binary_linux_wheel_py3.7_rocm5.2_smoke_test_pip
python_version: '3.7'
requires:
- nightly_binary_linux_wheel_py3.7_rocm5.1.1
- nightly_binary_linux_wheel_py3.7_rocm5.2
- binary_linux_wheel:
cuda_version: cpu
filters:
......@@ -2324,18 +2324,18 @@ workflows:
requires:
- nightly_binary_linux_wheel_py3.8_cu117
- binary_linux_wheel:
cuda_version: rocm5.0
cuda_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'
requires:
- build_ffmpeg_linux
wheel_docker_image: pytorch/manylinux-rocm:5.0
wheel_docker_image: pytorch/manylinux-rocm:5.1.1
- binary_wheel_upload:
context: org-member
filters:
......@@ -2344,35 +2344,35 @@ workflows:
- 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/
- smoke_test_linux_pip:
cuda_version: rocm5.0
cuda_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_smoke_test_pip
name: nightly_binary_linux_wheel_py3.8_rocm5.1.1_smoke_test_pip
python_version: '3.8'
requires:
- nightly_binary_linux_wheel_py3.8_rocm5.0
- nightly_binary_linux_wheel_py3.8_rocm5.1.1
- binary_linux_wheel:
cuda_version: rocm5.1.1
cuda_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'
requires:
- build_ffmpeg_linux
wheel_docker_image: pytorch/manylinux-rocm:5.1.1
wheel_docker_image: pytorch/manylinux-rocm:5.2
- binary_wheel_upload:
context: org-member
filters:
......@@ -2381,22 +2381,22 @@ workflows:
- 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/
- smoke_test_linux_pip:
cuda_version: rocm5.1.1
cuda_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_smoke_test_pip
name: nightly_binary_linux_wheel_py3.8_rocm5.2_smoke_test_pip
python_version: '3.8'
requires:
- nightly_binary_linux_wheel_py3.8_rocm5.1.1
- nightly_binary_linux_wheel_py3.8_rocm5.2
- binary_linux_wheel:
cuda_version: cpu
filters:
......@@ -2545,18 +2545,18 @@ workflows:
requires:
- nightly_binary_linux_wheel_py3.9_cu117
- binary_linux_wheel:
cuda_version: rocm5.0
cuda_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'
requires:
- build_ffmpeg_linux
wheel_docker_image: pytorch/manylinux-rocm:5.0
wheel_docker_image: pytorch/manylinux-rocm:5.1.1
- binary_wheel_upload:
context: org-member
filters:
......@@ -2565,35 +2565,35 @@ workflows:
- 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/
- smoke_test_linux_pip:
cuda_version: rocm5.0
cuda_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_smoke_test_pip
name: nightly_binary_linux_wheel_py3.9_rocm5.1.1_smoke_test_pip
python_version: '3.9'
requires:
- nightly_binary_linux_wheel_py3.9_rocm5.0
- nightly_binary_linux_wheel_py3.9_rocm5.1.1
- binary_linux_wheel:
cuda_version: rocm5.1.1
cuda_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'
requires:
- build_ffmpeg_linux
wheel_docker_image: pytorch/manylinux-rocm:5.1.1
wheel_docker_image: pytorch/manylinux-rocm:5.2
- binary_wheel_upload:
context: org-member
filters:
......@@ -2602,22 +2602,22 @@ workflows:
- 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/
- smoke_test_linux_pip:
cuda_version: rocm5.1.1
cuda_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_smoke_test_pip
name: nightly_binary_linux_wheel_py3.9_rocm5.2_smoke_test_pip
python_version: '3.9'
requires:
- nightly_binary_linux_wheel_py3.9_rocm5.1.1
- nightly_binary_linux_wheel_py3.9_rocm5.2
- binary_linux_wheel:
cuda_version: cpu
filters:
......@@ -2766,18 +2766,18 @@ workflows:
requires:
- nightly_binary_linux_wheel_py3.10_cu117
- binary_linux_wheel:
cuda_version: rocm5.0
cuda_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'
requires:
- build_ffmpeg_linux
wheel_docker_image: pytorch/manylinux-rocm:5.0
wheel_docker_image: pytorch/manylinux-rocm:5.1.1
- binary_wheel_upload:
context: org-member
filters:
......@@ -2786,35 +2786,35 @@ workflows:
- 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/
- smoke_test_linux_pip:
cuda_version: rocm5.0
cuda_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_smoke_test_pip
name: nightly_binary_linux_wheel_py3.10_rocm5.1.1_smoke_test_pip
python_version: '3.10'
requires:
- nightly_binary_linux_wheel_py3.10_rocm5.0
- nightly_binary_linux_wheel_py3.10_rocm5.1.1
- binary_linux_wheel:
cuda_version: rocm5.1.1
cuda_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'
requires:
- build_ffmpeg_linux
wheel_docker_image: pytorch/manylinux-rocm:5.1.1
wheel_docker_image: pytorch/manylinux-rocm:5.2
- binary_wheel_upload:
context: org-member
filters:
......@@ -2823,22 +2823,22 @@ workflows:
- 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/
- smoke_test_linux_pip:
cuda_version: rocm5.1.1
cuda_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_smoke_test_pip
name: nightly_binary_linux_wheel_py3.10_rocm5.2_smoke_test_pip
python_version: '3.10'
requires:
- nightly_binary_linux_wheel_py3.10_rocm5.1.1
- nightly_binary_linux_wheel_py3.10_rocm5.2
- binary_macos_wheel:
cuda_version: cpu
filters:
......
......@@ -23,7 +23,7 @@ from jinja2 import select_autoescape
PYTHON_VERSIONS = ["3.7", "3.8", "3.9", "3.10"]
CU_VERSIONS_DICT = {
"linux": ["cpu", "cu102", "cu116", "cu117", "rocm5.0", "rocm5.1.1"],
"linux": ["cpu", "cu102", "cu116", "cu117", "rocm5.1.1", "rocm5.2"],
"windows": ["cpu", "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