Skip to content

GitLab

Commit bb73934f authored by Nikita Shulga's avatar Nikita Shulga Committed by Facebook GitHub Bot
Browse files

Update ROCM version 4.1 -> 4.3.1 and 4.5 (#2186) 

Summary:
Also, retire cuda-10.2

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

Reviewed By: mthrok

Differential Revision: D33917595

Pulled By: malfet

fbshipit-source-id: 060d3fa706279fe45ffd1f4f99e5727520612d56
parent 157cb2a2
Hide whitespace changes
Inline Side-by-side
Showing with 132 additions and 261 deletions
.circleci/config.yml
View file @ bb73934f
...@@ -725,13 +725,6 @@ workflows: ...@@ -725,13 +725,6 @@ workflows:
python_version: '3.7' python_version: '3.7'
requires: requires:
- download_third_parties_nix - download_third_parties_nix
- binary_linux_wheel:
cuda_version: cu102
name: binary_linux_wheel_py3.7_cu102
python_version: '3.7'
requires:
- download_third_parties_nix
wheel_docker_image: pytorch/manylinux-cuda102
- binary_linux_wheel: - binary_linux_wheel:
cuda_version: cu111 cuda_version: cu111
name: binary_linux_wheel_py3.7_cu111 name: binary_linux_wheel_py3.7_cu111
...@@ -754,12 +747,19 @@ workflows: ...@@ -754,12 +747,19 @@ workflows:
- download_third_parties_nix - download_third_parties_nix
wheel_docker_image: pytorch/manylinux-cuda115 wheel_docker_image: pytorch/manylinux-cuda115
- binary_linux_wheel: - binary_linux_wheel:
cuda_version: rocm4.1 cuda_version: rocm4.3.1
name: binary_linux_wheel_py3.7_rocm4.1 name: binary_linux_wheel_py3.7_rocm4.3.1
python_version: '3.7'
requires:
- download_third_parties_nix
wheel_docker_image: pytorch/manylinux-rocm:4.3.1
- binary_linux_wheel:
cuda_version: rocm4.5.2
name: binary_linux_wheel_py3.7_rocm4.5.2
python_version: '3.7' python_version: '3.7'
requires: requires:
- download_third_parties_nix - download_third_parties_nix
wheel_docker_image: pytorch/manylinux-rocm:4.1 wheel_docker_image: pytorch/manylinux-rocm:4.5.2
- binary_linux_wheel: - binary_linux_wheel:
cuda_version: cpu cuda_version: cpu
filters: filters:
...@@ -772,13 +772,6 @@ workflows: ...@@ -772,13 +772,6 @@ workflows:
python_version: '3.8' python_version: '3.8'
requires: requires:
- download_third_parties_nix - download_third_parties_nix
- binary_linux_wheel:
cuda_version: cu102
name: binary_linux_wheel_py3.8_cu102
python_version: '3.8'
requires:
- download_third_parties_nix
wheel_docker_image: pytorch/manylinux-cuda102
- binary_linux_wheel: - binary_linux_wheel:
cuda_version: cu111 cuda_version: cu111
name: binary_linux_wheel_py3.8_cu111 name: binary_linux_wheel_py3.8_cu111
...@@ -801,25 +794,25 @@ workflows: ...@@ -801,25 +794,25 @@ workflows:
- download_third_parties_nix - download_third_parties_nix
wheel_docker_image: pytorch/manylinux-cuda115 wheel_docker_image: pytorch/manylinux-cuda115
- binary_linux_wheel: - binary_linux_wheel:
cuda_version: rocm4.1 cuda_version: rocm4.3.1
name: binary_linux_wheel_py3.8_rocm4.1 name: binary_linux_wheel_py3.8_rocm4.3.1
python_version: '3.8' python_version: '3.8'
requires: requires:
- download_third_parties_nix - download_third_parties_nix
wheel_docker_image: pytorch/manylinux-rocm:4.1 wheel_docker_image: pytorch/manylinux-rocm:4.3.1
- binary_linux_wheel: - binary_linux_wheel:
cuda_version: cpu cuda_version: rocm4.5.2
name: binary_linux_wheel_py3.9_cpu name: binary_linux_wheel_py3.8_rocm4.5.2
python_version: '3.9' python_version: '3.8'
requires: requires:
- download_third_parties_nix - download_third_parties_nix
wheel_docker_image: pytorch/manylinux-rocm:4.5.2
- binary_linux_wheel: - binary_linux_wheel:
cuda_version: cu102 cuda_version: cpu
name: binary_linux_wheel_py3.9_cu102 name: binary_linux_wheel_py3.9_cpu
python_version: '3.9' python_version: '3.9'
requires: requires:
- download_third_parties_nix - download_third_parties_nix
wheel_docker_image: pytorch/manylinux-cuda102
- binary_linux_wheel: - binary_linux_wheel:
cuda_version: cu111 cuda_version: cu111
name: binary_linux_wheel_py3.9_cu111 name: binary_linux_wheel_py3.9_cu111
...@@ -842,12 +835,19 @@ workflows: ...@@ -842,12 +835,19 @@ workflows:
- download_third_parties_nix - download_third_parties_nix
wheel_docker_image: pytorch/manylinux-cuda115 wheel_docker_image: pytorch/manylinux-cuda115
- binary_linux_wheel: - binary_linux_wheel:
cuda_version: rocm4.1 cuda_version: rocm4.3.1
name: binary_linux_wheel_py3.9_rocm4.1 name: binary_linux_wheel_py3.9_rocm4.3.1
python_version: '3.9' python_version: '3.9'
requires: requires:
- download_third_parties_nix - download_third_parties_nix
wheel_docker_image: pytorch/manylinux-rocm:4.1 wheel_docker_image: pytorch/manylinux-rocm:4.3.1
- binary_linux_wheel:
cuda_version: rocm4.5.2
name: binary_linux_wheel_py3.9_rocm4.5.2
python_version: '3.9'
requires:
- download_third_parties_nix
wheel_docker_image: pytorch/manylinux-rocm:4.5.2
- 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
...@@ -915,13 +915,6 @@ workflows: ...@@ -915,13 +915,6 @@ workflows:
python_version: '3.7' python_version: '3.7'
requires: requires:
- download_third_parties_nix - download_third_parties_nix
- binary_linux_conda:
conda_docker_image: pytorch/conda-builder:cuda102
cuda_version: cu102
name: binary_linux_conda_py3.7_cu102
python_version: '3.7'
requires:
- download_third_parties_nix
- binary_linux_conda: - binary_linux_conda:
conda_docker_image: pytorch/conda-builder:cuda111 conda_docker_image: pytorch/conda-builder:cuda111
cuda_version: cu111 cuda_version: cu111
...@@ -950,13 +943,6 @@ workflows: ...@@ -950,13 +943,6 @@ workflows:
python_version: '3.8' python_version: '3.8'
requires: requires:
- download_third_parties_nix - download_third_parties_nix
- binary_linux_conda:
conda_docker_image: pytorch/conda-builder:cuda102
cuda_version: cu102
name: binary_linux_conda_py3.8_cu102
python_version: '3.8'
requires:
- download_third_parties_nix
- binary_linux_conda: - binary_linux_conda:
conda_docker_image: pytorch/conda-builder:cuda111 conda_docker_image: pytorch/conda-builder:cuda111
cuda_version: cu111 cuda_version: cu111
...@@ -985,13 +971,6 @@ workflows: ...@@ -985,13 +971,6 @@ workflows:
python_version: '3.9' python_version: '3.9'
requires: requires:
- download_third_parties_nix - download_third_parties_nix
- binary_linux_conda:
conda_docker_image: pytorch/conda-builder:cuda102
cuda_version: cu102
name: binary_linux_conda_py3.9_cu102
python_version: '3.9'
requires:
- download_third_parties_nix
- binary_linux_conda: - binary_linux_conda:
conda_docker_image: pytorch/conda-builder:cuda111 conda_docker_image: pytorch/conda-builder:cuda111
cuda_version: cu111 cuda_version: cu111
...@@ -1243,43 +1222,6 @@ workflows: ...@@ -1243,43 +1222,6 @@ workflows:
python_version: '3.7' python_version: '3.7'
requires: requires:
- nightly_binary_linux_wheel_py3.7_cpu_upload - nightly_binary_linux_wheel_py3.7_cpu_upload
- binary_linux_wheel:
cuda_version: cu102
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.7_cu102
python_version: '3.7'
requires:
- download_third_parties_nix
wheel_docker_image: pytorch/manylinux-cuda102
- binary_wheel_upload:
context: org-member
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.7_cu102_upload
requires:
- nightly_binary_linux_wheel_py3.7_cu102
subfolder: cu102/
- smoke_test_linux_pip:
cuda_version: cu102
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.7_cu102_smoke_test_pip
python_version: '3.7'
requires:
- nightly_binary_linux_wheel_py3.7_cu102_upload
- binary_linux_wheel: - binary_linux_wheel:
cuda_version: cu111 cuda_version: cu111
filters: filters:
...@@ -1392,18 +1334,18 @@ workflows: ...@@ -1392,18 +1334,18 @@ workflows:
requires: requires:
- nightly_binary_linux_wheel_py3.7_cu115_upload - nightly_binary_linux_wheel_py3.7_cu115_upload
- binary_linux_wheel: - binary_linux_wheel:
cuda_version: rocm4.1 cuda_version: rocm4.3.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.1 name: nightly_binary_linux_wheel_py3.7_rocm4.3.1
python_version: '3.7' python_version: '3.7'
requires: requires:
- download_third_parties_nix - download_third_parties_nix
wheel_docker_image: pytorch/manylinux-rocm:4.1 wheel_docker_image: pytorch/manylinux-rocm:4.3.1
- binary_wheel_upload: - binary_wheel_upload:
context: org-member context: org-member
filters: filters:
...@@ -1412,34 +1354,35 @@ workflows: ...@@ -1412,34 +1354,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.1_upload name: nightly_binary_linux_wheel_py3.7_rocm4.3.1_upload
requires: requires:
- nightly_binary_linux_wheel_py3.7_rocm4.1 - nightly_binary_linux_wheel_py3.7_rocm4.3.1
subfolder: rocm4.1/ subfolder: rocm4.3.1/
- smoke_test_linux_pip: - smoke_test_linux_pip:
cuda_version: rocm4.1 cuda_version: rocm4.3.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.1_smoke_test_pip name: nightly_binary_linux_wheel_py3.7_rocm4.3.1_smoke_test_pip
python_version: '3.7' python_version: '3.7'
requires: requires:
- nightly_binary_linux_wheel_py3.7_rocm4.1_upload - nightly_binary_linux_wheel_py3.7_rocm4.3.1_upload
- binary_linux_wheel: - binary_linux_wheel:
cuda_version: cpu cuda_version: rocm4.5.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_cpu name: nightly_binary_linux_wheel_py3.7_rocm4.5.2
python_version: '3.8' python_version: '3.7'
requires: requires:
- download_third_parties_nix - download_third_parties_nix
wheel_docker_image: pytorch/manylinux-rocm:4.5.2
- binary_wheel_upload: - binary_wheel_upload:
context: org-member context: org-member
filters: filters:
...@@ -1448,35 +1391,34 @@ workflows: ...@@ -1448,35 +1391,34 @@ 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_cpu_upload name: nightly_binary_linux_wheel_py3.7_rocm4.5.2_upload
requires: requires:
- nightly_binary_linux_wheel_py3.8_cpu - nightly_binary_linux_wheel_py3.7_rocm4.5.2
subfolder: cpu/ subfolder: rocm4.5.2/
- smoke_test_linux_pip: - smoke_test_linux_pip:
cuda_version: cpu cuda_version: rocm4.5.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_cpu_smoke_test_pip name: nightly_binary_linux_wheel_py3.7_rocm4.5.2_smoke_test_pip
python_version: '3.8' python_version: '3.7'
requires: requires:
- nightly_binary_linux_wheel_py3.8_cpu_upload - nightly_binary_linux_wheel_py3.7_rocm4.5.2_upload
- binary_linux_wheel: - binary_linux_wheel:
cuda_version: cu102 cuda_version: cpu
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_cu102 name: nightly_binary_linux_wheel_py3.8_cpu
python_version: '3.8' python_version: '3.8'
requires: requires:
- download_third_parties_nix - download_third_parties_nix
wheel_docker_image: pytorch/manylinux-cuda102
- binary_wheel_upload: - binary_wheel_upload:
context: org-member context: org-member
filters: filters:
...@@ -1485,22 +1427,22 @@ workflows: ...@@ -1485,22 +1427,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_cu102_upload name: nightly_binary_linux_wheel_py3.8_cpu_upload
requires: requires:
- nightly_binary_linux_wheel_py3.8_cu102 - nightly_binary_linux_wheel_py3.8_cpu
subfolder: cu102/ subfolder: cpu/
- smoke_test_linux_pip: - smoke_test_linux_pip:
cuda_version: cu102 cuda_version: cpu
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_cu102_smoke_test_pip name: nightly_binary_linux_wheel_py3.8_cpu_smoke_test_pip
python_version: '3.8' python_version: '3.8'
requires: requires:
- nightly_binary_linux_wheel_py3.8_cu102_upload - nightly_binary_linux_wheel_py3.8_cpu_upload
- binary_linux_wheel: - binary_linux_wheel:
cuda_version: cu111 cuda_version: cu111
filters: filters:
...@@ -1613,18 +1555,18 @@ workflows: ...@@ -1613,18 +1555,18 @@ workflows:
requires: requires:
- nightly_binary_linux_wheel_py3.8_cu115_upload - nightly_binary_linux_wheel_py3.8_cu115_upload
- binary_linux_wheel: - binary_linux_wheel:
cuda_version: rocm4.1 cuda_version: rocm4.3.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.1 name: nightly_binary_linux_wheel_py3.8_rocm4.3.1
python_version: '3.8' python_version: '3.8'
requires: requires:
- download_third_parties_nix - download_third_parties_nix
wheel_docker_image: pytorch/manylinux-rocm:4.1 wheel_docker_image: pytorch/manylinux-rocm:4.3.1
- binary_wheel_upload: - binary_wheel_upload:
context: org-member context: org-member
filters: filters:
...@@ -1633,34 +1575,35 @@ workflows: ...@@ -1633,34 +1575,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.1_upload name: nightly_binary_linux_wheel_py3.8_rocm4.3.1_upload
requires: requires:
- nightly_binary_linux_wheel_py3.8_rocm4.1 - nightly_binary_linux_wheel_py3.8_rocm4.3.1
subfolder: rocm4.1/ subfolder: rocm4.3.1/
- smoke_test_linux_pip: - smoke_test_linux_pip:
cuda_version: rocm4.1 cuda_version: rocm4.3.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.1_smoke_test_pip name: nightly_binary_linux_wheel_py3.8_rocm4.3.1_smoke_test_pip
python_version: '3.8' python_version: '3.8'
requires: requires:
- nightly_binary_linux_wheel_py3.8_rocm4.1_upload - nightly_binary_linux_wheel_py3.8_rocm4.3.1_upload
- binary_linux_wheel: - binary_linux_wheel:
cuda_version: cpu cuda_version: rocm4.5.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_cpu name: nightly_binary_linux_wheel_py3.8_rocm4.5.2
python_version: '3.9' python_version: '3.8'
requires: requires:
- download_third_parties_nix - download_third_parties_nix
wheel_docker_image: pytorch/manylinux-rocm:4.5.2
- binary_wheel_upload: - binary_wheel_upload:
context: org-member context: org-member
filters: filters:
...@@ -1669,35 +1612,34 @@ workflows: ...@@ -1669,35 +1612,34 @@ 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_cpu_upload name: nightly_binary_linux_wheel_py3.8_rocm4.5.2_upload
requires: requires:
- nightly_binary_linux_wheel_py3.9_cpu - nightly_binary_linux_wheel_py3.8_rocm4.5.2
subfolder: cpu/ subfolder: rocm4.5.2/
- smoke_test_linux_pip: - smoke_test_linux_pip:
cuda_version: cpu cuda_version: rocm4.5.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_cpu_smoke_test_pip name: nightly_binary_linux_wheel_py3.8_rocm4.5.2_smoke_test_pip
python_version: '3.9' python_version: '3.8'
requires: requires:
- nightly_binary_linux_wheel_py3.9_cpu_upload - nightly_binary_linux_wheel_py3.8_rocm4.5.2_upload
- binary_linux_wheel: - binary_linux_wheel:
cuda_version: cu102 cuda_version: cpu
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_cu102 name: nightly_binary_linux_wheel_py3.9_cpu
python_version: '3.9' python_version: '3.9'
requires: requires:
- download_third_parties_nix - download_third_parties_nix
wheel_docker_image: pytorch/manylinux-cuda102
- binary_wheel_upload: - binary_wheel_upload:
context: org-member context: org-member
filters: filters:
...@@ -1706,22 +1648,22 @@ workflows: ...@@ -1706,22 +1648,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_cu102_upload name: nightly_binary_linux_wheel_py3.9_cpu_upload
requires: requires:
- nightly_binary_linux_wheel_py3.9_cu102 - nightly_binary_linux_wheel_py3.9_cpu
subfolder: cu102/ subfolder: cpu/
- smoke_test_linux_pip: - smoke_test_linux_pip:
cuda_version: cu102 cuda_version: cpu
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_cu102_smoke_test_pip name: nightly_binary_linux_wheel_py3.9_cpu_smoke_test_pip
python_version: '3.9' python_version: '3.9'
requires: requires:
- nightly_binary_linux_wheel_py3.9_cu102_upload - nightly_binary_linux_wheel_py3.9_cpu_upload
- binary_linux_wheel: - binary_linux_wheel:
cuda_version: cu111 cuda_version: cu111
filters: filters:
...@@ -1834,18 +1776,55 @@ workflows: ...@@ -1834,18 +1776,55 @@ workflows:
requires: requires:
- nightly_binary_linux_wheel_py3.9_cu115_upload - nightly_binary_linux_wheel_py3.9_cu115_upload
- binary_linux_wheel: - binary_linux_wheel:
cuda_version: rocm4.1 cuda_version: rocm4.3.1
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.9_rocm4.3.1
python_version: '3.9'
requires:
- download_third_parties_nix
wheel_docker_image: pytorch/manylinux-rocm:4.3.1
- binary_wheel_upload:
context: org-member
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.9_rocm4.3.1_upload
requires:
- nightly_binary_linux_wheel_py3.9_rocm4.3.1
subfolder: rocm4.3.1/
- smoke_test_linux_pip:
cuda_version: rocm4.3.1
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.9_rocm4.3.1_smoke_test_pip
python_version: '3.9'
requires:
- nightly_binary_linux_wheel_py3.9_rocm4.3.1_upload
- binary_linux_wheel:
cuda_version: rocm4.5.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_rocm4.1 name: nightly_binary_linux_wheel_py3.9_rocm4.5.2
python_version: '3.9' python_version: '3.9'
requires: requires:
- download_third_parties_nix - download_third_parties_nix
wheel_docker_image: pytorch/manylinux-rocm:4.1 wheel_docker_image: pytorch/manylinux-rocm:4.5.2
- binary_wheel_upload: - binary_wheel_upload:
context: org-member context: org-member
filters: filters:
...@@ -1854,22 +1833,22 @@ workflows: ...@@ -1854,22 +1833,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.1_upload name: nightly_binary_linux_wheel_py3.9_rocm4.5.2_upload
requires: requires:
- nightly_binary_linux_wheel_py3.9_rocm4.1 - nightly_binary_linux_wheel_py3.9_rocm4.5.2
subfolder: rocm4.1/ subfolder: rocm4.5.2/
- smoke_test_linux_pip: - smoke_test_linux_pip:
cuda_version: rocm4.1 cuda_version: rocm4.5.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_rocm4.1_smoke_test_pip name: nightly_binary_linux_wheel_py3.9_rocm4.5.2_smoke_test_pip
python_version: '3.9' python_version: '3.9'
requires: requires:
- nightly_binary_linux_wheel_py3.9_rocm4.1_upload - nightly_binary_linux_wheel_py3.9_rocm4.5.2_upload
- binary_macos_wheel: - binary_macos_wheel:
cuda_version: cpu cuda_version: cpu
filters: filters:
...@@ -2290,42 +2269,6 @@ workflows: ...@@ -2290,42 +2269,6 @@ workflows:
python_version: '3.7' python_version: '3.7'
requires: requires:
- nightly_binary_linux_conda_py3.7_cpu_upload - nightly_binary_linux_conda_py3.7_cpu_upload
- binary_linux_conda:
conda_docker_image: pytorch/conda-builder:cuda102
cuda_version: cu102
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_conda_py3.7_cu102
python_version: '3.7'
requires:
- download_third_parties_nix
- binary_conda_upload:
context: org-member
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_conda_py3.7_cu102_upload
requires:
- nightly_binary_linux_conda_py3.7_cu102
- smoke_test_linux_conda_gpu:
cuda_version: cu102
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_conda_py3.7_cu102_smoke_test_conda
python_version: '3.7'
requires:
- nightly_binary_linux_conda_py3.7_cu102_upload
- binary_linux_conda: - binary_linux_conda:
conda_docker_image: pytorch/conda-builder:cuda111 conda_docker_image: pytorch/conda-builder:cuda111
cuda_version: cu111 cuda_version: cu111
...@@ -2470,42 +2413,6 @@ workflows: ...@@ -2470,42 +2413,6 @@ workflows:
python_version: '3.8' python_version: '3.8'
requires: requires:
- nightly_binary_linux_conda_py3.8_cpu_upload - nightly_binary_linux_conda_py3.8_cpu_upload
- binary_linux_conda:
conda_docker_image: pytorch/conda-builder:cuda102
cuda_version: cu102
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_conda_py3.8_cu102
python_version: '3.8'
requires:
- download_third_parties_nix
- binary_conda_upload:
context: org-member
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_conda_py3.8_cu102_upload
requires:
- nightly_binary_linux_conda_py3.8_cu102
- smoke_test_linux_conda_gpu:
cuda_version: cu102
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_conda_py3.8_cu102_smoke_test_conda
python_version: '3.8'
requires:
- nightly_binary_linux_conda_py3.8_cu102_upload
- binary_linux_conda: - binary_linux_conda:
conda_docker_image: pytorch/conda-builder:cuda111 conda_docker_image: pytorch/conda-builder:cuda111
cuda_version: cu111 cuda_version: cu111
...@@ -2650,42 +2557,6 @@ workflows: ...@@ -2650,42 +2557,6 @@ workflows:
python_version: '3.9' python_version: '3.9'
requires: requires:
- nightly_binary_linux_conda_py3.9_cpu_upload - nightly_binary_linux_conda_py3.9_cpu_upload
- binary_linux_conda:
conda_docker_image: pytorch/conda-builder:cuda102
cuda_version: cu102
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_conda_py3.9_cu102
python_version: '3.9'
requires:
- download_third_parties_nix
- binary_conda_upload:
context: org-member
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_conda_py3.9_cu102_upload
requires:
- nightly_binary_linux_conda_py3.9_cu102
- smoke_test_linux_conda_gpu:
cuda_version: cu102
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_conda_py3.9_cu102_smoke_test_conda
python_version: '3.9'
requires:
- nightly_binary_linux_conda_py3.9_cu102_upload
- binary_linux_conda: - binary_linux_conda:
conda_docker_image: pytorch/conda-builder:cuda111 conda_docker_image: pytorch/conda-builder:cuda111
cuda_version: cu111 cuda_version: cu111
......
.circleci/regenerate.py
View file @ bb73934f
...@@ -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"] PYTHON_VERSIONS = ["3.7", "3.8", "3.9"]
CU_VERSIONS_DICT = { CU_VERSIONS_DICT = {
"linux": ["cpu", "cu102", "cu111", "cu113", "cu115", "rocm4.1"], "linux": ["cpu", "cu111", "cu113", "cu115", "rocm4.3.1", "rocm4.5.2"],
"windows": ["cpu", "cu113", "cu115"], "windows": ["cpu", "cu113", "cu115"],
"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