Skip to content

GitLab

Commit eab2f39d authored by Kyle Chen's avatar Kyle Chen Committed by Facebook GitHub Bot
Browse files

[ROCm] Update to rocm5.1.1 (#2362) 

Summary:
previous update for rocm: https://github.com/pytorch/audio/pull/2186

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

Reviewed By: seemethere

Differential Revision: D36283672

Pulled By: mthrok

fbshipit-source-id: bfd38940d027c8ccd72ab48991e5ab7f84b0e9c0
parent 4b021ae3
Show whitespace changes
Inline Side-by-side
Showing with 121 additions and 121 deletions
.circleci/config.yml
View file @ eab2f39d
...@@ -938,31 +938,31 @@ workflows: ...@@ -938,31 +938,31 @@ workflows:
requires: requires:
- binary_linux_wheel_py3.7_cu116 - binary_linux_wheel_py3.7_cu116
- binary_linux_wheel: - binary_linux_wheel:
cuda_version: rocm4.3.1 cuda_version: rocm5.0
name: binary_linux_wheel_py3.7_rocm4.3.1 name: binary_linux_wheel_py3.7_rocm5.0
python_version: '3.7' python_version: '3.7'
requires: requires:
- build_ffmpeg_linux - build_ffmpeg_linux
wheel_docker_image: pytorch/manylinux-rocm:4.3.1 wheel_docker_image: pytorch/manylinux-rocm:5.0
- smoke_test_linux_pip: - smoke_test_linux_pip:
cuda_version: rocm4.3.1 cuda_version: rocm5.0
name: binary_linux_wheel_py3.7_rocm4.3.1_smoke_test_pip name: binary_linux_wheel_py3.7_rocm5.0_smoke_test_pip
python_version: '3.7' python_version: '3.7'
requires: requires:
- binary_linux_wheel_py3.7_rocm4.3.1 - binary_linux_wheel_py3.7_rocm5.0
- binary_linux_wheel: - binary_linux_wheel:
cuda_version: rocm4.5.2 cuda_version: rocm5.1.1
name: binary_linux_wheel_py3.7_rocm4.5.2 name: binary_linux_wheel_py3.7_rocm5.1.1
python_version: '3.7' python_version: '3.7'
requires: requires:
- build_ffmpeg_linux - build_ffmpeg_linux
wheel_docker_image: pytorch/manylinux-rocm:4.5.2 wheel_docker_image: pytorch/manylinux-rocm:5.1.1
- smoke_test_linux_pip: - smoke_test_linux_pip:
cuda_version: rocm4.5.2 cuda_version: rocm5.1.1
name: binary_linux_wheel_py3.7_rocm4.5.2_smoke_test_pip name: binary_linux_wheel_py3.7_rocm5.1.1_smoke_test_pip
python_version: '3.7' python_version: '3.7'
requires: requires:
- binary_linux_wheel_py3.7_rocm4.5.2 - binary_linux_wheel_py3.7_rocm5.1.1
- binary_linux_wheel: - binary_linux_wheel:
cuda_version: cpu cuda_version: cpu
filters: filters:
...@@ -1027,31 +1027,31 @@ workflows: ...@@ -1027,31 +1027,31 @@ workflows:
requires: requires:
- binary_linux_wheel_py3.8_cu116 - binary_linux_wheel_py3.8_cu116
- binary_linux_wheel: - binary_linux_wheel:
cuda_version: rocm4.3.1 cuda_version: rocm5.0
name: binary_linux_wheel_py3.8_rocm4.3.1 name: binary_linux_wheel_py3.8_rocm5.0
python_version: '3.8' python_version: '3.8'
requires: requires:
- build_ffmpeg_linux - build_ffmpeg_linux
wheel_docker_image: pytorch/manylinux-rocm:4.3.1 wheel_docker_image: pytorch/manylinux-rocm:5.0
- smoke_test_linux_pip: - smoke_test_linux_pip:
cuda_version: rocm4.3.1 cuda_version: rocm5.0
name: binary_linux_wheel_py3.8_rocm4.3.1_smoke_test_pip name: binary_linux_wheel_py3.8_rocm5.0_smoke_test_pip
python_version: '3.8' python_version: '3.8'
requires: requires:
- binary_linux_wheel_py3.8_rocm4.3.1 - binary_linux_wheel_py3.8_rocm5.0
- binary_linux_wheel: - binary_linux_wheel:
cuda_version: rocm4.5.2 cuda_version: rocm5.1.1
name: binary_linux_wheel_py3.8_rocm4.5.2 name: binary_linux_wheel_py3.8_rocm5.1.1
python_version: '3.8' python_version: '3.8'
requires: requires:
- build_ffmpeg_linux - build_ffmpeg_linux
wheel_docker_image: pytorch/manylinux-rocm:4.5.2 wheel_docker_image: pytorch/manylinux-rocm:5.1.1
- smoke_test_linux_pip: - smoke_test_linux_pip:
cuda_version: rocm4.5.2 cuda_version: rocm5.1.1
name: binary_linux_wheel_py3.8_rocm4.5.2_smoke_test_pip name: binary_linux_wheel_py3.8_rocm5.1.1_smoke_test_pip
python_version: '3.8' python_version: '3.8'
requires: requires:
- binary_linux_wheel_py3.8_rocm4.5.2 - binary_linux_wheel_py3.8_rocm5.1.1
- 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
...@@ -1104,31 +1104,31 @@ workflows: ...@@ -1104,31 +1104,31 @@ workflows:
requires: requires:
- binary_linux_wheel_py3.9_cu116 - binary_linux_wheel_py3.9_cu116
- binary_linux_wheel: - binary_linux_wheel:
cuda_version: rocm4.3.1 cuda_version: rocm5.0
name: binary_linux_wheel_py3.9_rocm4.3.1 name: binary_linux_wheel_py3.9_rocm5.0
python_version: '3.9' python_version: '3.9'
requires: requires:
- build_ffmpeg_linux - build_ffmpeg_linux
wheel_docker_image: pytorch/manylinux-rocm:4.3.1 wheel_docker_image: pytorch/manylinux-rocm:5.0
- smoke_test_linux_pip: - smoke_test_linux_pip:
cuda_version: rocm4.3.1 cuda_version: rocm5.0
name: binary_linux_wheel_py3.9_rocm4.3.1_smoke_test_pip name: binary_linux_wheel_py3.9_rocm5.0_smoke_test_pip
python_version: '3.9' python_version: '3.9'
requires: requires:
- binary_linux_wheel_py3.9_rocm4.3.1 - binary_linux_wheel_py3.9_rocm5.0
- binary_linux_wheel: - binary_linux_wheel:
cuda_version: rocm4.5.2 cuda_version: rocm5.1.1
name: binary_linux_wheel_py3.9_rocm4.5.2 name: binary_linux_wheel_py3.9_rocm5.1.1
python_version: '3.9' python_version: '3.9'
requires: requires:
- build_ffmpeg_linux - build_ffmpeg_linux
wheel_docker_image: pytorch/manylinux-rocm:4.5.2 wheel_docker_image: pytorch/manylinux-rocm:5.1.1
- smoke_test_linux_pip: - smoke_test_linux_pip:
cuda_version: rocm4.5.2 cuda_version: rocm5.1.1
name: binary_linux_wheel_py3.9_rocm4.5.2_smoke_test_pip name: binary_linux_wheel_py3.9_rocm5.1.1_smoke_test_pip
python_version: '3.9' python_version: '3.9'
requires: requires:
- binary_linux_wheel_py3.9_rocm4.5.2 - binary_linux_wheel_py3.9_rocm5.1.1
- 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
...@@ -1181,31 +1181,31 @@ workflows: ...@@ -1181,31 +1181,31 @@ workflows:
requires: requires:
- binary_linux_wheel_py3.10_cu116 - binary_linux_wheel_py3.10_cu116
- binary_linux_wheel: - binary_linux_wheel:
cuda_version: rocm4.3.1 cuda_version: rocm5.0
name: binary_linux_wheel_py3.10_rocm4.3.1 name: binary_linux_wheel_py3.10_rocm5.0
python_version: '3.10' python_version: '3.10'
requires: requires:
- build_ffmpeg_linux - build_ffmpeg_linux
wheel_docker_image: pytorch/manylinux-rocm:4.3.1 wheel_docker_image: pytorch/manylinux-rocm:5.0
- smoke_test_linux_pip: - smoke_test_linux_pip:
cuda_version: rocm4.3.1 cuda_version: rocm5.0
name: binary_linux_wheel_py3.10_rocm4.3.1_smoke_test_pip name: binary_linux_wheel_py3.10_rocm5.0_smoke_test_pip
python_version: '3.10' python_version: '3.10'
requires: requires:
- binary_linux_wheel_py3.10_rocm4.3.1 - binary_linux_wheel_py3.10_rocm5.0
- binary_linux_wheel: - binary_linux_wheel:
cuda_version: rocm4.5.2 cuda_version: rocm5.1.1
name: binary_linux_wheel_py3.10_rocm4.5.2 name: binary_linux_wheel_py3.10_rocm5.1.1
python_version: '3.10' python_version: '3.10'
requires: requires:
- build_ffmpeg_linux - build_ffmpeg_linux
wheel_docker_image: pytorch/manylinux-rocm:4.5.2 wheel_docker_image: pytorch/manylinux-rocm:5.1.1
- smoke_test_linux_pip: - smoke_test_linux_pip:
cuda_version: rocm4.5.2 cuda_version: rocm5.1.1
name: binary_linux_wheel_py3.10_rocm4.5.2_smoke_test_pip name: binary_linux_wheel_py3.10_rocm5.1.1_smoke_test_pip
python_version: '3.10' python_version: '3.10'
requires: requires:
- binary_linux_wheel_py3.10_rocm4.5.2 - binary_linux_wheel_py3.10_rocm5.1.1
- 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
...@@ -2071,18 +2071,18 @@ workflows: ...@@ -2071,18 +2071,18 @@ workflows:
requires: requires:
- nightly_binary_linux_wheel_py3.7_cu116 - nightly_binary_linux_wheel_py3.7_cu116
- binary_linux_wheel: - binary_linux_wheel:
cuda_version: rocm4.3.1 cuda_version: rocm5.0
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_rocm4.3.1 name: nightly_binary_linux_wheel_py3.7_rocm5.0
python_version: '3.7' python_version: '3.7'
requires: requires:
- build_ffmpeg_linux - build_ffmpeg_linux
wheel_docker_image: pytorch/manylinux-rocm:4.3.1 wheel_docker_image: pytorch/manylinux-rocm:5.0
- binary_wheel_upload: - binary_wheel_upload:
context: org-member context: org-member
filters: filters:
...@@ -2091,35 +2091,35 @@ workflows: ...@@ -2091,35 +2091,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_rocm4.3.1_upload name: nightly_binary_linux_wheel_py3.7_rocm5.0_upload
requires: requires:
- nightly_binary_linux_wheel_py3.7_rocm4.3.1 - nightly_binary_linux_wheel_py3.7_rocm5.0
subfolder: rocm4.3.1/ subfolder: rocm5.0/
- smoke_test_linux_pip: - smoke_test_linux_pip:
cuda_version: rocm4.3.1 cuda_version: rocm5.0
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_rocm4.3.1_smoke_test_pip name: nightly_binary_linux_wheel_py3.7_rocm5.0_smoke_test_pip
python_version: '3.7' python_version: '3.7'
requires: requires:
- nightly_binary_linux_wheel_py3.7_rocm4.3.1 - nightly_binary_linux_wheel_py3.7_rocm5.0
- binary_linux_wheel: - binary_linux_wheel:
cuda_version: rocm4.5.2 cuda_version: rocm5.1.1
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_rocm4.5.2 name: nightly_binary_linux_wheel_py3.7_rocm5.1.1
python_version: '3.7' python_version: '3.7'
requires: requires:
- build_ffmpeg_linux - build_ffmpeg_linux
wheel_docker_image: pytorch/manylinux-rocm:4.5.2 wheel_docker_image: pytorch/manylinux-rocm:5.1.1
- binary_wheel_upload: - binary_wheel_upload:
context: org-member context: org-member
filters: filters:
...@@ -2128,22 +2128,22 @@ workflows: ...@@ -2128,22 +2128,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_rocm4.5.2_upload name: nightly_binary_linux_wheel_py3.7_rocm5.1.1_upload
requires: requires:
- nightly_binary_linux_wheel_py3.7_rocm4.5.2 - nightly_binary_linux_wheel_py3.7_rocm5.1.1
subfolder: rocm4.5.2/ subfolder: rocm5.1.1/
- smoke_test_linux_pip: - smoke_test_linux_pip:
cuda_version: rocm4.5.2 cuda_version: rocm5.1.1
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_rocm4.5.2_smoke_test_pip name: nightly_binary_linux_wheel_py3.7_rocm5.1.1_smoke_test_pip
python_version: '3.7' python_version: '3.7'
requires: requires:
- nightly_binary_linux_wheel_py3.7_rocm4.5.2 - nightly_binary_linux_wheel_py3.7_rocm5.1.1
- binary_linux_wheel: - binary_linux_wheel:
cuda_version: cpu cuda_version: cpu
filters: filters:
...@@ -2292,18 +2292,18 @@ workflows: ...@@ -2292,18 +2292,18 @@ workflows:
requires: requires:
- nightly_binary_linux_wheel_py3.8_cu116 - nightly_binary_linux_wheel_py3.8_cu116
- binary_linux_wheel: - binary_linux_wheel:
cuda_version: rocm4.3.1 cuda_version: rocm5.0
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_rocm4.3.1 name: nightly_binary_linux_wheel_py3.8_rocm5.0
python_version: '3.8' python_version: '3.8'
requires: requires:
- build_ffmpeg_linux - build_ffmpeg_linux
wheel_docker_image: pytorch/manylinux-rocm:4.3.1 wheel_docker_image: pytorch/manylinux-rocm:5.0
- binary_wheel_upload: - binary_wheel_upload:
context: org-member context: org-member
filters: filters:
...@@ -2312,35 +2312,35 @@ workflows: ...@@ -2312,35 +2312,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_rocm4.3.1_upload name: nightly_binary_linux_wheel_py3.8_rocm5.0_upload
requires: requires:
- nightly_binary_linux_wheel_py3.8_rocm4.3.1 - nightly_binary_linux_wheel_py3.8_rocm5.0
subfolder: rocm4.3.1/ subfolder: rocm5.0/
- smoke_test_linux_pip: - smoke_test_linux_pip:
cuda_version: rocm4.3.1 cuda_version: rocm5.0
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_rocm4.3.1_smoke_test_pip name: nightly_binary_linux_wheel_py3.8_rocm5.0_smoke_test_pip
python_version: '3.8' python_version: '3.8'
requires: requires:
- nightly_binary_linux_wheel_py3.8_rocm4.3.1 - nightly_binary_linux_wheel_py3.8_rocm5.0
- binary_linux_wheel: - binary_linux_wheel:
cuda_version: rocm4.5.2 cuda_version: rocm5.1.1
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_rocm4.5.2 name: nightly_binary_linux_wheel_py3.8_rocm5.1.1
python_version: '3.8' python_version: '3.8'
requires: requires:
- build_ffmpeg_linux - build_ffmpeg_linux
wheel_docker_image: pytorch/manylinux-rocm:4.5.2 wheel_docker_image: pytorch/manylinux-rocm:5.1.1
- binary_wheel_upload: - binary_wheel_upload:
context: org-member context: org-member
filters: filters:
...@@ -2349,22 +2349,22 @@ workflows: ...@@ -2349,22 +2349,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_rocm4.5.2_upload name: nightly_binary_linux_wheel_py3.8_rocm5.1.1_upload
requires: requires:
- nightly_binary_linux_wheel_py3.8_rocm4.5.2 - nightly_binary_linux_wheel_py3.8_rocm5.1.1
subfolder: rocm4.5.2/ subfolder: rocm5.1.1/
- smoke_test_linux_pip: - smoke_test_linux_pip:
cuda_version: rocm4.5.2 cuda_version: rocm5.1.1
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_rocm4.5.2_smoke_test_pip name: nightly_binary_linux_wheel_py3.8_rocm5.1.1_smoke_test_pip
python_version: '3.8' python_version: '3.8'
requires: requires:
- nightly_binary_linux_wheel_py3.8_rocm4.5.2 - nightly_binary_linux_wheel_py3.8_rocm5.1.1
- binary_linux_wheel: - binary_linux_wheel:
cuda_version: cpu cuda_version: cpu
filters: filters:
...@@ -2513,18 +2513,18 @@ workflows: ...@@ -2513,18 +2513,18 @@ workflows:
requires: requires:
- nightly_binary_linux_wheel_py3.9_cu116 - nightly_binary_linux_wheel_py3.9_cu116
- binary_linux_wheel: - binary_linux_wheel:
cuda_version: rocm4.3.1 cuda_version: rocm5.0
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_rocm4.3.1 name: nightly_binary_linux_wheel_py3.9_rocm5.0
python_version: '3.9' python_version: '3.9'
requires: requires:
- build_ffmpeg_linux - build_ffmpeg_linux
wheel_docker_image: pytorch/manylinux-rocm:4.3.1 wheel_docker_image: pytorch/manylinux-rocm:5.0
- binary_wheel_upload: - binary_wheel_upload:
context: org-member context: org-member
filters: filters:
...@@ -2533,35 +2533,35 @@ workflows: ...@@ -2533,35 +2533,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_rocm4.3.1_upload name: nightly_binary_linux_wheel_py3.9_rocm5.0_upload
requires: requires:
- nightly_binary_linux_wheel_py3.9_rocm4.3.1 - nightly_binary_linux_wheel_py3.9_rocm5.0
subfolder: rocm4.3.1/ subfolder: rocm5.0/
- smoke_test_linux_pip: - smoke_test_linux_pip:
cuda_version: rocm4.3.1 cuda_version: rocm5.0
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_rocm4.3.1_smoke_test_pip name: nightly_binary_linux_wheel_py3.9_rocm5.0_smoke_test_pip
python_version: '3.9' python_version: '3.9'
requires: requires:
- nightly_binary_linux_wheel_py3.9_rocm4.3.1 - nightly_binary_linux_wheel_py3.9_rocm5.0
- binary_linux_wheel: - binary_linux_wheel:
cuda_version: rocm4.5.2 cuda_version: rocm5.1.1
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_rocm4.5.2 name: nightly_binary_linux_wheel_py3.9_rocm5.1.1
python_version: '3.9' python_version: '3.9'
requires: requires:
- build_ffmpeg_linux - build_ffmpeg_linux
wheel_docker_image: pytorch/manylinux-rocm:4.5.2 wheel_docker_image: pytorch/manylinux-rocm:5.1.1
- binary_wheel_upload: - binary_wheel_upload:
context: org-member context: org-member
filters: filters:
...@@ -2570,22 +2570,22 @@ workflows: ...@@ -2570,22 +2570,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_rocm4.5.2_upload name: nightly_binary_linux_wheel_py3.9_rocm5.1.1_upload
requires: requires:
- nightly_binary_linux_wheel_py3.9_rocm4.5.2 - nightly_binary_linux_wheel_py3.9_rocm5.1.1
subfolder: rocm4.5.2/ subfolder: rocm5.1.1/
- smoke_test_linux_pip: - smoke_test_linux_pip:
cuda_version: rocm4.5.2 cuda_version: rocm5.1.1
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_rocm4.5.2_smoke_test_pip name: nightly_binary_linux_wheel_py3.9_rocm5.1.1_smoke_test_pip
python_version: '3.9' python_version: '3.9'
requires: requires:
- nightly_binary_linux_wheel_py3.9_rocm4.5.2 - nightly_binary_linux_wheel_py3.9_rocm5.1.1
- binary_linux_wheel: - binary_linux_wheel:
cuda_version: cpu cuda_version: cpu
filters: filters:
...@@ -2734,18 +2734,18 @@ workflows: ...@@ -2734,18 +2734,18 @@ workflows:
requires: requires:
- nightly_binary_linux_wheel_py3.10_cu116 - nightly_binary_linux_wheel_py3.10_cu116
- binary_linux_wheel: - binary_linux_wheel:
cuda_version: rocm4.3.1 cuda_version: rocm5.0
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_rocm4.3.1 name: nightly_binary_linux_wheel_py3.10_rocm5.0
python_version: '3.10' python_version: '3.10'
requires: requires:
- build_ffmpeg_linux - build_ffmpeg_linux
wheel_docker_image: pytorch/manylinux-rocm:4.3.1 wheel_docker_image: pytorch/manylinux-rocm:5.0
- binary_wheel_upload: - binary_wheel_upload:
context: org-member context: org-member
filters: filters:
...@@ -2754,35 +2754,35 @@ workflows: ...@@ -2754,35 +2754,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_rocm4.3.1_upload name: nightly_binary_linux_wheel_py3.10_rocm5.0_upload
requires: requires:
- nightly_binary_linux_wheel_py3.10_rocm4.3.1 - nightly_binary_linux_wheel_py3.10_rocm5.0
subfolder: rocm4.3.1/ subfolder: rocm5.0/
- smoke_test_linux_pip: - smoke_test_linux_pip:
cuda_version: rocm4.3.1 cuda_version: rocm5.0
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_rocm4.3.1_smoke_test_pip name: nightly_binary_linux_wheel_py3.10_rocm5.0_smoke_test_pip
python_version: '3.10' python_version: '3.10'
requires: requires:
- nightly_binary_linux_wheel_py3.10_rocm4.3.1 - nightly_binary_linux_wheel_py3.10_rocm5.0
- binary_linux_wheel: - binary_linux_wheel:
cuda_version: rocm4.5.2 cuda_version: rocm5.1.1
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_rocm4.5.2 name: nightly_binary_linux_wheel_py3.10_rocm5.1.1
python_version: '3.10' python_version: '3.10'
requires: requires:
- build_ffmpeg_linux - build_ffmpeg_linux
wheel_docker_image: pytorch/manylinux-rocm:4.5.2 wheel_docker_image: pytorch/manylinux-rocm:5.1.1
- binary_wheel_upload: - binary_wheel_upload:
context: org-member context: org-member
filters: filters:
...@@ -2791,22 +2791,22 @@ workflows: ...@@ -2791,22 +2791,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_rocm4.5.2_upload name: nightly_binary_linux_wheel_py3.10_rocm5.1.1_upload
requires: requires:
- nightly_binary_linux_wheel_py3.10_rocm4.5.2 - nightly_binary_linux_wheel_py3.10_rocm5.1.1
subfolder: rocm4.5.2/ subfolder: rocm5.1.1/
- smoke_test_linux_pip: - smoke_test_linux_pip:
cuda_version: rocm4.5.2 cuda_version: rocm5.1.1
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_rocm4.5.2_smoke_test_pip name: nightly_binary_linux_wheel_py3.10_rocm5.1.1_smoke_test_pip
python_version: '3.10' python_version: '3.10'
requires: requires:
- nightly_binary_linux_wheel_py3.10_rocm4.5.2 - nightly_binary_linux_wheel_py3.10_rocm5.1.1
- binary_macos_wheel: - binary_macos_wheel:
cuda_version: cpu cuda_version: cpu
filters: filters:
......
.circleci/regenerate.py
View file @ eab2f39d
...@@ -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", "cu102", "cu113", "cu116", "rocm4.3.1", "rocm4.5.2"], "linux": ["cpu", "cu102", "cu113", "cu116", "rocm5.0", "rocm5.1.1"],
"windows": ["cpu", "cu113", "cu116"], "windows": ["cpu", "cu113", "cu116"],
"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