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

Upgrade nightly wheels to ROCm5.3 (#2853)

Summary:
Dependent on PR https://github.com/pytorch/pytorch/pull/89101

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

Reviewed By: atalman, osalpekar

Differential Revision: D41737634

Pulled By: malfet

fbshipit-source-id: 715a97a2da8ef309cea78d971b47c07463495683
parent d234498c
...@@ -968,31 +968,31 @@ workflows: ...@@ -968,31 +968,31 @@ workflows:
requires: requires:
- binary_linux_wheel_py3.7_cu117 - binary_linux_wheel_py3.7_cu117
- binary_linux_wheel: - binary_linux_wheel:
cuda_version: rocm5.1.1 cuda_version: rocm5.2
name: binary_linux_wheel_py3.7_rocm5.1.1 name: binary_linux_wheel_py3.7_rocm5.2
python_version: '3.7' python_version: '3.7'
requires: requires:
- build_ffmpeg_linux - build_ffmpeg_linux
wheel_docker_image: pytorch/manylinux-rocm:5.1.1 wheel_docker_image: pytorch/manylinux-rocm:5.2
- smoke_test_linux_pip: - smoke_test_linux_pip:
cuda_version: rocm5.1.1 cuda_version: rocm5.2
name: binary_linux_wheel_py3.7_rocm5.1.1_smoke_test_pip name: binary_linux_wheel_py3.7_rocm5.2_smoke_test_pip
python_version: '3.7' python_version: '3.7'
requires: requires:
- binary_linux_wheel_py3.7_rocm5.1.1 - binary_linux_wheel_py3.7_rocm5.2
- binary_linux_wheel: - binary_linux_wheel:
cuda_version: rocm5.2 cuda_version: rocm5.3
name: binary_linux_wheel_py3.7_rocm5.2 name: binary_linux_wheel_py3.7_rocm5.3
python_version: '3.7' python_version: '3.7'
requires: requires:
- build_ffmpeg_linux - build_ffmpeg_linux
wheel_docker_image: pytorch/manylinux-rocm:5.2 wheel_docker_image: pytorch/manylinux-rocm:5.3
- smoke_test_linux_pip: - smoke_test_linux_pip:
cuda_version: rocm5.2 cuda_version: rocm5.3
name: binary_linux_wheel_py3.7_rocm5.2_smoke_test_pip name: binary_linux_wheel_py3.7_rocm5.3_smoke_test_pip
python_version: '3.7' python_version: '3.7'
requires: requires:
- binary_linux_wheel_py3.7_rocm5.2 - binary_linux_wheel_py3.7_rocm5.3
- binary_linux_wheel: - binary_linux_wheel:
cuda_version: cpu cuda_version: cpu
filters: filters:
...@@ -1044,31 +1044,31 @@ workflows: ...@@ -1044,31 +1044,31 @@ workflows:
requires: requires:
- binary_linux_wheel_py3.8_cu117 - binary_linux_wheel_py3.8_cu117
- binary_linux_wheel: - binary_linux_wheel:
cuda_version: rocm5.1.1 cuda_version: rocm5.2
name: binary_linux_wheel_py3.8_rocm5.1.1 name: binary_linux_wheel_py3.8_rocm5.2
python_version: '3.8' python_version: '3.8'
requires: requires:
- build_ffmpeg_linux - build_ffmpeg_linux
wheel_docker_image: pytorch/manylinux-rocm:5.1.1 wheel_docker_image: pytorch/manylinux-rocm:5.2
- smoke_test_linux_pip: - smoke_test_linux_pip:
cuda_version: rocm5.1.1 cuda_version: rocm5.2
name: binary_linux_wheel_py3.8_rocm5.1.1_smoke_test_pip name: binary_linux_wheel_py3.8_rocm5.2_smoke_test_pip
python_version: '3.8' python_version: '3.8'
requires: requires:
- binary_linux_wheel_py3.8_rocm5.1.1 - binary_linux_wheel_py3.8_rocm5.2
- binary_linux_wheel: - binary_linux_wheel:
cuda_version: rocm5.2 cuda_version: rocm5.3
name: binary_linux_wheel_py3.8_rocm5.2 name: binary_linux_wheel_py3.8_rocm5.3
python_version: '3.8' python_version: '3.8'
requires: requires:
- build_ffmpeg_linux - build_ffmpeg_linux
wheel_docker_image: pytorch/manylinux-rocm:5.2 wheel_docker_image: pytorch/manylinux-rocm:5.3
- smoke_test_linux_pip: - smoke_test_linux_pip:
cuda_version: rocm5.2 cuda_version: rocm5.3
name: binary_linux_wheel_py3.8_rocm5.2_smoke_test_pip name: binary_linux_wheel_py3.8_rocm5.3_smoke_test_pip
python_version: '3.8' python_version: '3.8'
requires: requires:
- binary_linux_wheel_py3.8_rocm5.2 - binary_linux_wheel_py3.8_rocm5.3
- binary_linux_wheel: - binary_linux_wheel:
cuda_version: cpu cuda_version: cpu
name: binary_linux_wheel_py3.9_cpu name: binary_linux_wheel_py3.9_cpu
...@@ -1108,31 +1108,31 @@ workflows: ...@@ -1108,31 +1108,31 @@ workflows:
requires: requires:
- binary_linux_wheel_py3.9_cu117 - binary_linux_wheel_py3.9_cu117
- binary_linux_wheel: - binary_linux_wheel:
cuda_version: rocm5.1.1 cuda_version: rocm5.2
name: binary_linux_wheel_py3.9_rocm5.1.1 name: binary_linux_wheel_py3.9_rocm5.2
python_version: '3.9' python_version: '3.9'
requires: requires:
- build_ffmpeg_linux - build_ffmpeg_linux
wheel_docker_image: pytorch/manylinux-rocm:5.1.1 wheel_docker_image: pytorch/manylinux-rocm:5.2
- smoke_test_linux_pip: - smoke_test_linux_pip:
cuda_version: rocm5.1.1 cuda_version: rocm5.2
name: binary_linux_wheel_py3.9_rocm5.1.1_smoke_test_pip name: binary_linux_wheel_py3.9_rocm5.2_smoke_test_pip
python_version: '3.9' python_version: '3.9'
requires: requires:
- binary_linux_wheel_py3.9_rocm5.1.1 - binary_linux_wheel_py3.9_rocm5.2
- binary_linux_wheel: - binary_linux_wheel:
cuda_version: rocm5.2 cuda_version: rocm5.3
name: binary_linux_wheel_py3.9_rocm5.2 name: binary_linux_wheel_py3.9_rocm5.3
python_version: '3.9' python_version: '3.9'
requires: requires:
- build_ffmpeg_linux - build_ffmpeg_linux
wheel_docker_image: pytorch/manylinux-rocm:5.2 wheel_docker_image: pytorch/manylinux-rocm:5.3
- smoke_test_linux_pip: - smoke_test_linux_pip:
cuda_version: rocm5.2 cuda_version: rocm5.3
name: binary_linux_wheel_py3.9_rocm5.2_smoke_test_pip name: binary_linux_wheel_py3.9_rocm5.3_smoke_test_pip
python_version: '3.9' python_version: '3.9'
requires: requires:
- binary_linux_wheel_py3.9_rocm5.2 - binary_linux_wheel_py3.9_rocm5.3
- binary_linux_wheel: - binary_linux_wheel:
cuda_version: cpu cuda_version: cpu
name: binary_linux_wheel_py3.10_cpu name: binary_linux_wheel_py3.10_cpu
...@@ -1172,31 +1172,31 @@ workflows: ...@@ -1172,31 +1172,31 @@ workflows:
requires: requires:
- binary_linux_wheel_py3.10_cu117 - binary_linux_wheel_py3.10_cu117
- binary_linux_wheel: - binary_linux_wheel:
cuda_version: rocm5.1.1 cuda_version: rocm5.2
name: binary_linux_wheel_py3.10_rocm5.1.1 name: binary_linux_wheel_py3.10_rocm5.2
python_version: '3.10' python_version: '3.10'
requires: requires:
- build_ffmpeg_linux - build_ffmpeg_linux
wheel_docker_image: pytorch/manylinux-rocm:5.1.1 wheel_docker_image: pytorch/manylinux-rocm:5.2
- smoke_test_linux_pip: - smoke_test_linux_pip:
cuda_version: rocm5.1.1 cuda_version: rocm5.2
name: binary_linux_wheel_py3.10_rocm5.1.1_smoke_test_pip name: binary_linux_wheel_py3.10_rocm5.2_smoke_test_pip
python_version: '3.10' python_version: '3.10'
requires: requires:
- binary_linux_wheel_py3.10_rocm5.1.1 - binary_linux_wheel_py3.10_rocm5.2
- binary_linux_wheel: - binary_linux_wheel:
cuda_version: rocm5.2 cuda_version: rocm5.3
name: binary_linux_wheel_py3.10_rocm5.2 name: binary_linux_wheel_py3.10_rocm5.3
python_version: '3.10' python_version: '3.10'
requires: requires:
- build_ffmpeg_linux - build_ffmpeg_linux
wheel_docker_image: pytorch/manylinux-rocm:5.2 wheel_docker_image: pytorch/manylinux-rocm:5.3
- smoke_test_linux_pip: - smoke_test_linux_pip:
cuda_version: rocm5.2 cuda_version: rocm5.3
name: binary_linux_wheel_py3.10_rocm5.2_smoke_test_pip name: binary_linux_wheel_py3.10_rocm5.3_smoke_test_pip
python_version: '3.10' python_version: '3.10'
requires: requires:
- binary_linux_wheel_py3.10_rocm5.2 - binary_linux_wheel_py3.10_rocm5.3
- binary_macos_wheel: - binary_macos_wheel:
cuda_version: cpu cuda_version: cpu
name: binary_macos_wheel_py3.7_cpu name: binary_macos_wheel_py3.7_cpu
...@@ -1974,18 +1974,18 @@ workflows: ...@@ -1974,18 +1974,18 @@ workflows:
requires: requires:
- nightly_binary_linux_wheel_py3.7_cu117 - nightly_binary_linux_wheel_py3.7_cu117
- binary_linux_wheel: - binary_linux_wheel:
cuda_version: rocm5.1.1 cuda_version: rocm5.2
filters: filters:
branches: branches:
only: only:
- nightly - 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'
requires: requires:
- build_ffmpeg_linux - build_ffmpeg_linux
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:
...@@ -1994,35 +1994,35 @@ workflows: ...@@ -1994,35 +1994,35 @@ workflows:
- nightly - 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/
- smoke_test_linux_pip: - smoke_test_linux_pip:
cuda_version: rocm5.1.1 cuda_version: rocm5.2
filters: filters:
branches: branches:
only: only:
- nightly - 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_smoke_test_pip name: nightly_binary_linux_wheel_py3.7_rocm5.2_smoke_test_pip
python_version: '3.7' python_version: '3.7'
requires: requires:
- nightly_binary_linux_wheel_py3.7_rocm5.1.1 - nightly_binary_linux_wheel_py3.7_rocm5.2
- binary_linux_wheel: - binary_linux_wheel:
cuda_version: rocm5.2 cuda_version: rocm5.3
filters: filters:
branches: branches:
only: only:
- nightly - 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'
requires: requires:
- build_ffmpeg_linux - build_ffmpeg_linux
wheel_docker_image: pytorch/manylinux-rocm:5.2 wheel_docker_image: pytorch/manylinux-rocm:5.3
- binary_wheel_upload: - binary_wheel_upload:
context: org-member context: org-member
filters: filters:
...@@ -2031,22 +2031,22 @@ workflows: ...@@ -2031,22 +2031,22 @@ workflows:
- nightly - 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_upload name: nightly_binary_linux_wheel_py3.7_rocm5.3_upload
requires: requires:
- nightly_binary_linux_wheel_py3.7_rocm5.2 - nightly_binary_linux_wheel_py3.7_rocm5.3
subfolder: rocm5.2/ subfolder: rocm5.3/
- smoke_test_linux_pip: - smoke_test_linux_pip:
cuda_version: rocm5.2 cuda_version: rocm5.3
filters: filters:
branches: branches:
only: only:
- nightly - 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_smoke_test_pip name: nightly_binary_linux_wheel_py3.7_rocm5.3_smoke_test_pip
python_version: '3.7' python_version: '3.7'
requires: requires:
- nightly_binary_linux_wheel_py3.7_rocm5.2 - nightly_binary_linux_wheel_py3.7_rocm5.3
- binary_linux_wheel: - binary_linux_wheel:
cuda_version: cpu cuda_version: cpu
filters: filters:
...@@ -2158,18 +2158,18 @@ workflows: ...@@ -2158,18 +2158,18 @@ workflows:
requires: requires:
- nightly_binary_linux_wheel_py3.8_cu117 - nightly_binary_linux_wheel_py3.8_cu117
- binary_linux_wheel: - binary_linux_wheel:
cuda_version: rocm5.1.1 cuda_version: rocm5.2
filters: filters:
branches: branches:
only: only:
- nightly - 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'
requires: requires:
- build_ffmpeg_linux - build_ffmpeg_linux
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:
...@@ -2178,35 +2178,35 @@ workflows: ...@@ -2178,35 +2178,35 @@ workflows:
- nightly - 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/
- smoke_test_linux_pip: - smoke_test_linux_pip:
cuda_version: rocm5.1.1 cuda_version: rocm5.2
filters: filters:
branches: branches:
only: only:
- nightly - 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_smoke_test_pip name: nightly_binary_linux_wheel_py3.8_rocm5.2_smoke_test_pip
python_version: '3.8' python_version: '3.8'
requires: requires:
- nightly_binary_linux_wheel_py3.8_rocm5.1.1 - nightly_binary_linux_wheel_py3.8_rocm5.2
- binary_linux_wheel: - binary_linux_wheel:
cuda_version: rocm5.2 cuda_version: rocm5.3
filters: filters:
branches: branches:
only: only:
- nightly - 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.2 name: nightly_binary_linux_wheel_py3.8_rocm5.3
python_version: '3.8' python_version: '3.8'
requires: requires:
- build_ffmpeg_linux - build_ffmpeg_linux
wheel_docker_image: pytorch/manylinux-rocm:5.2 wheel_docker_image: pytorch/manylinux-rocm:5.3
- binary_wheel_upload: - binary_wheel_upload:
context: org-member context: org-member
filters: filters:
...@@ -2215,22 +2215,22 @@ workflows: ...@@ -2215,22 +2215,22 @@ workflows:
- nightly - 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.2_upload name: nightly_binary_linux_wheel_py3.8_rocm5.3_upload
requires: requires:
- nightly_binary_linux_wheel_py3.8_rocm5.2 - nightly_binary_linux_wheel_py3.8_rocm5.3
subfolder: rocm5.2/ subfolder: rocm5.3/
- smoke_test_linux_pip: - smoke_test_linux_pip:
cuda_version: rocm5.2 cuda_version: rocm5.3
filters: filters:
branches: branches:
only: only:
- nightly - 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.2_smoke_test_pip name: nightly_binary_linux_wheel_py3.8_rocm5.3_smoke_test_pip
python_version: '3.8' python_version: '3.8'
requires: requires:
- nightly_binary_linux_wheel_py3.8_rocm5.2 - nightly_binary_linux_wheel_py3.8_rocm5.3
- binary_linux_wheel: - binary_linux_wheel:
cuda_version: cpu cuda_version: cpu
filters: filters:
...@@ -2342,18 +2342,18 @@ workflows: ...@@ -2342,18 +2342,18 @@ workflows:
requires: requires:
- nightly_binary_linux_wheel_py3.9_cu117 - nightly_binary_linux_wheel_py3.9_cu117
- binary_linux_wheel: - binary_linux_wheel:
cuda_version: rocm5.1.1 cuda_version: rocm5.2
filters: filters:
branches: branches:
only: only:
- nightly - 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'
requires: requires:
- build_ffmpeg_linux - build_ffmpeg_linux
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:
...@@ -2362,35 +2362,35 @@ workflows: ...@@ -2362,35 +2362,35 @@ workflows:
- nightly - 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/
- smoke_test_linux_pip: - smoke_test_linux_pip:
cuda_version: rocm5.1.1 cuda_version: rocm5.2
filters: filters:
branches: branches:
only: only:
- nightly - 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_smoke_test_pip name: nightly_binary_linux_wheel_py3.9_rocm5.2_smoke_test_pip
python_version: '3.9' python_version: '3.9'
requires: requires:
- nightly_binary_linux_wheel_py3.9_rocm5.1.1 - nightly_binary_linux_wheel_py3.9_rocm5.2
- binary_linux_wheel: - binary_linux_wheel:
cuda_version: rocm5.2 cuda_version: rocm5.3
filters: filters:
branches: branches:
only: only:
- nightly - 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.2 name: nightly_binary_linux_wheel_py3.9_rocm5.3
python_version: '3.9' python_version: '3.9'
requires: requires:
- build_ffmpeg_linux - build_ffmpeg_linux
wheel_docker_image: pytorch/manylinux-rocm:5.2 wheel_docker_image: pytorch/manylinux-rocm:5.3
- binary_wheel_upload: - binary_wheel_upload:
context: org-member context: org-member
filters: filters:
...@@ -2399,22 +2399,22 @@ workflows: ...@@ -2399,22 +2399,22 @@ workflows:
- nightly - 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.2_upload name: nightly_binary_linux_wheel_py3.9_rocm5.3_upload
requires: requires:
- nightly_binary_linux_wheel_py3.9_rocm5.2 - nightly_binary_linux_wheel_py3.9_rocm5.3
subfolder: rocm5.2/ subfolder: rocm5.3/
- smoke_test_linux_pip: - smoke_test_linux_pip:
cuda_version: rocm5.2 cuda_version: rocm5.3
filters: filters:
branches: branches:
only: only:
- nightly - 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.2_smoke_test_pip name: nightly_binary_linux_wheel_py3.9_rocm5.3_smoke_test_pip
python_version: '3.9' python_version: '3.9'
requires: requires:
- nightly_binary_linux_wheel_py3.9_rocm5.2 - nightly_binary_linux_wheel_py3.9_rocm5.3
- binary_linux_wheel: - binary_linux_wheel:
cuda_version: cpu cuda_version: cpu
filters: filters:
...@@ -2526,18 +2526,18 @@ workflows: ...@@ -2526,18 +2526,18 @@ workflows:
requires: requires:
- nightly_binary_linux_wheel_py3.10_cu117 - nightly_binary_linux_wheel_py3.10_cu117
- binary_linux_wheel: - binary_linux_wheel:
cuda_version: rocm5.1.1 cuda_version: rocm5.2
filters: filters:
branches: branches:
only: only:
- nightly - 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'
requires: requires:
- build_ffmpeg_linux - build_ffmpeg_linux
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:
...@@ -2546,35 +2546,35 @@ workflows: ...@@ -2546,35 +2546,35 @@ workflows:
- nightly - 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/
- smoke_test_linux_pip: - smoke_test_linux_pip:
cuda_version: rocm5.1.1 cuda_version: rocm5.2
filters: filters:
branches: branches:
only: only:
- nightly - 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_smoke_test_pip name: nightly_binary_linux_wheel_py3.10_rocm5.2_smoke_test_pip
python_version: '3.10' python_version: '3.10'
requires: requires:
- nightly_binary_linux_wheel_py3.10_rocm5.1.1 - nightly_binary_linux_wheel_py3.10_rocm5.2
- binary_linux_wheel: - binary_linux_wheel:
cuda_version: rocm5.2 cuda_version: rocm5.3
filters: filters:
branches: branches:
only: only:
- nightly - 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.2 name: nightly_binary_linux_wheel_py3.10_rocm5.3
python_version: '3.10' python_version: '3.10'
requires: requires:
- build_ffmpeg_linux - build_ffmpeg_linux
wheel_docker_image: pytorch/manylinux-rocm:5.2 wheel_docker_image: pytorch/manylinux-rocm:5.3
- binary_wheel_upload: - binary_wheel_upload:
context: org-member context: org-member
filters: filters:
...@@ -2583,22 +2583,22 @@ workflows: ...@@ -2583,22 +2583,22 @@ workflows:
- nightly - 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.2_upload name: nightly_binary_linux_wheel_py3.10_rocm5.3_upload
requires: requires:
- nightly_binary_linux_wheel_py3.10_rocm5.2 - nightly_binary_linux_wheel_py3.10_rocm5.3
subfolder: rocm5.2/ subfolder: rocm5.3/
- smoke_test_linux_pip: - smoke_test_linux_pip:
cuda_version: rocm5.2 cuda_version: rocm5.3
filters: filters:
branches: branches:
only: only:
- nightly - 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.2_smoke_test_pip name: nightly_binary_linux_wheel_py3.10_rocm5.3_smoke_test_pip
python_version: '3.10' python_version: '3.10'
requires: requires:
- nightly_binary_linux_wheel_py3.10_rocm5.2 - nightly_binary_linux_wheel_py3.10_rocm5.3
- binary_macos_wheel: - binary_macos_wheel:
cuda_version: cpu cuda_version: cpu
filters: filters:
......
...@@ -23,7 +23,7 @@ from jinja2 import select_autoescape ...@@ -23,7 +23,7 @@ from jinja2 import select_autoescape
PYTHON_VERSIONS = ["3.7", "3.8", "3.9", "3.10"] PYTHON_VERSIONS = ["3.7", "3.8", "3.9", "3.10"]
CU_VERSIONS_DICT = { CU_VERSIONS_DICT = {
"linux": ["cpu", "cu116", "cu117", "rocm5.1.1", "rocm5.2"], "linux": ["cpu", "cu116", "cu117", "rocm5.2", "rocm5.3"],
"windows": ["cpu", "cu116", "cu117"], "windows": ["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