Skip to content

GitLab

Commit 27dff6ba authored by Andrey Talman's avatar Andrey Talman Committed by Facebook GitHub Bot
Browse files

Add Python 3.10 (build and test) (#2224) 

Summary:
Adding py3.10 to audio

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

Reviewed By: malfet, atalman, mthrok

Differential Revision: D34442377

Pulled By: seemethere

fbshipit-source-id: 2656de73427063958d609a74c01b526a476cb06a
parent 1fb10077
Hide whitespace changes
Inline Side-by-side
Showing with 1053 additions and 124 deletions
.circleci/config.yml
View file @ 27dff6ba
...@@ -475,6 +475,7 @@ jobs: ...@@ -475,6 +475,7 @@ jobs:
command: .circleci/unittest/linux/scripts/run_test.sh command: .circleci/unittest/linux/scripts/run_test.sh
environment: environment:
TORCHAUDIO_TEST_ALLOW_SKIP_IF_NO_CUDA: true TORCHAUDIO_TEST_ALLOW_SKIP_IF_NO_CUDA: true
TORCHAUDIO_TEST_ALLOW_SKIP_IF_ON_PYTHON_310: true
- store_test_results: - store_test_results:
path: test-results path: test-results
- store_artifacts: - store_artifacts:
...@@ -504,7 +505,10 @@ jobs: ...@@ -504,7 +505,10 @@ jobs:
command: docker run -t --gpus all -e UPLOAD_CHANNEL -e CONDA_CHANNEL_FLAGS -e BUILD_FFMPEG=1 -v $PWD:$PWD -w $PWD "${image_name}" .circleci/unittest/linux/scripts/install.sh command: docker run -t --gpus all -e UPLOAD_CHANNEL -e CONDA_CHANNEL_FLAGS -e BUILD_FFMPEG=1 -v $PWD:$PWD -w $PWD "${image_name}" .circleci/unittest/linux/scripts/install.sh
- run: - run:
name: Run tests name: Run tests
command: docker run -t --gpus all -v $PWD:$PWD -w $PWD -e "CI=${CI}" "${image_name}" .circleci/unittest/linux/scripts/run_test.sh environment:
TORCHAUDIO_TEST_ALLOW_SKIP_IF_ON_PYTHON_310: true
command: |
docker run -t --gpus all -v $PWD:$PWD -w $PWD -e "CI=${CI}" -e TORCHAUDIO_TEST_ALLOW_SKIP_IF_ON_PYTHON_310 "${image_name}" .circleci/unittest/linux/scripts/run_test.sh
- store_test_results: - store_test_results:
path: test-results path: test-results
- store_artifacts: - store_artifacts:
...@@ -540,6 +544,8 @@ jobs: ...@@ -540,6 +544,8 @@ jobs:
TORCHAUDIO_TEST_ALLOW_SKIP_IF_NO_CUDA: true TORCHAUDIO_TEST_ALLOW_SKIP_IF_NO_CUDA: true
TORCHAUDIO_TEST_ALLOW_SKIP_IF_NO_KALDI: true TORCHAUDIO_TEST_ALLOW_SKIP_IF_NO_KALDI: true
TORCHAUDIO_TEST_ALLOW_SKIP_IF_NO_SOX: true TORCHAUDIO_TEST_ALLOW_SKIP_IF_NO_SOX: true
TORCHAUDIO_TEST_ALLOW_SKIP_IF_ON_PYTHON_310: true
TORCHAUDIO_TEST_ALLOW_SKIP_IF_NO_MOD_sentencepiece: true
- store_test_results: - store_test_results:
path: test-results path: test-results
- store_artifacts: - store_artifacts:
...@@ -583,6 +589,8 @@ jobs: ...@@ -583,6 +589,8 @@ jobs:
TORCHAUDIO_TEST_ALLOW_SKIP_IF_NO_CTC_DECODER: true TORCHAUDIO_TEST_ALLOW_SKIP_IF_NO_CTC_DECODER: true
TORCHAUDIO_TEST_ALLOW_SKIP_IF_NO_KALDI: true TORCHAUDIO_TEST_ALLOW_SKIP_IF_NO_KALDI: true
TORCHAUDIO_TEST_ALLOW_SKIP_IF_NO_SOX: true TORCHAUDIO_TEST_ALLOW_SKIP_IF_NO_SOX: true
TORCHAUDIO_TEST_ALLOW_SKIP_IF_ON_PYTHON_310: true
TORCHAUDIO_TEST_ALLOW_SKIP_IF_NO_MOD_sentencepiece: true
- store_test_results: - store_test_results:
path: test-results path: test-results
- store_artifacts: - store_artifacts:
...@@ -618,6 +626,8 @@ jobs: ...@@ -618,6 +626,8 @@ jobs:
TORCHAUDIO_TEST_ALLOW_SKIP_IF_NO_CMD_COMPUTE_SPECTROGRAM_FEATS: true TORCHAUDIO_TEST_ALLOW_SKIP_IF_NO_CMD_COMPUTE_SPECTROGRAM_FEATS: true
TORCHAUDIO_TEST_ALLOW_SKIP_IF_NO_CUDA: true TORCHAUDIO_TEST_ALLOW_SKIP_IF_NO_CUDA: true
TORCHAUDIO_TEST_ALLOW_SKIP_IF_NO_QUANTIZATION: true TORCHAUDIO_TEST_ALLOW_SKIP_IF_NO_QUANTIZATION: true
TORCHAUDIO_TEST_ALLOW_SKIP_IF_ON_PYTHON_310: true
TORCHAUDIO_TEST_ALLOW_SKIP_IF_NO_MOD_sentencepiece: true
- store_test_results: - store_test_results:
path: test-results path: test-results
- store_artifacts: - store_artifacts:
...@@ -869,6 +879,54 @@ workflows: ...@@ -869,6 +879,54 @@ workflows:
requires: requires:
- download_third_parties_nix - download_third_parties_nix
wheel_docker_image: pytorch/manylinux-rocm:4.5.2 wheel_docker_image: pytorch/manylinux-rocm:4.5.2
- binary_linux_wheel:
cuda_version: cpu
name: binary_linux_wheel_py3.10_cpu
python_version: '3.10'
requires:
- download_third_parties_nix
- binary_linux_wheel:
cuda_version: cu102
name: binary_linux_wheel_py3.10_cu102
python_version: '3.10'
requires:
- download_third_parties_nix
wheel_docker_image: pytorch/manylinux-cuda102
- binary_linux_wheel:
cuda_version: cu111
name: binary_linux_wheel_py3.10_cu111
python_version: '3.10'
requires:
- download_third_parties_nix
wheel_docker_image: pytorch/manylinux-cuda111
- binary_linux_wheel:
cuda_version: cu113
name: binary_linux_wheel_py3.10_cu113
python_version: '3.10'
requires:
- download_third_parties_nix
wheel_docker_image: pytorch/manylinux-cuda113
- binary_linux_wheel:
cuda_version: cu115
name: binary_linux_wheel_py3.10_cu115
python_version: '3.10'
requires:
- download_third_parties_nix
wheel_docker_image: pytorch/manylinux-cuda115
- binary_linux_wheel:
cuda_version: rocm4.3.1
name: binary_linux_wheel_py3.10_rocm4.3.1
python_version: '3.10'
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.10_rocm4.5.2
python_version: '3.10'
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
...@@ -887,6 +945,12 @@ workflows: ...@@ -887,6 +945,12 @@ workflows:
python_version: '3.9' python_version: '3.9'
requires: requires:
- download_third_parties_nix - download_third_parties_nix
- binary_macos_wheel:
cuda_version: cpu
name: binary_macos_wheel_py3.10_cpu
python_version: '3.10'
requires:
- download_third_parties_nix
- binary_windows_wheel: - binary_windows_wheel:
cuda_version: cpu cuda_version: cpu
name: binary_windows_wheel_py3.7_cpu name: binary_windows_wheel_py3.7_cpu
...@@ -929,6 +993,20 @@ workflows: ...@@ -929,6 +993,20 @@ workflows:
name: binary_windows_wheel_py3.9_cu115 name: binary_windows_wheel_py3.9_cu115
python_version: '3.9' python_version: '3.9'
wheel_docker_image: pytorch/manylinux-cuda115 wheel_docker_image: pytorch/manylinux-cuda115
- binary_windows_wheel:
cuda_version: cpu
name: binary_windows_wheel_py3.10_cpu
python_version: '3.10'
- binary_windows_wheel:
cuda_version: cu113
name: binary_windows_wheel_py3.10_cu113
python_version: '3.10'
wheel_docker_image: pytorch/manylinux-cuda113
- binary_windows_wheel:
cuda_version: cu115
name: binary_windows_wheel_py3.10_cu115
python_version: '3.10'
wheel_docker_image: pytorch/manylinux-cuda115
- binary_linux_conda: - binary_linux_conda:
conda_docker_image: pytorch/conda-builder:cpu conda_docker_image: pytorch/conda-builder:cpu
cuda_version: cpu cuda_version: cpu
...@@ -1034,6 +1112,41 @@ workflows: ...@@ -1034,6 +1112,41 @@ 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:cpu
cuda_version: cpu
name: binary_linux_conda_py3.10_cpu
python_version: '3.10'
requires:
- download_third_parties_nix
- binary_linux_conda:
conda_docker_image: pytorch/conda-builder:cuda102
cuda_version: cu102
name: binary_linux_conda_py3.10_cu102
python_version: '3.10'
requires:
- download_third_parties_nix
- binary_linux_conda:
conda_docker_image: pytorch/conda-builder:cuda111
cuda_version: cu111
name: binary_linux_conda_py3.10_cu111
python_version: '3.10'
requires:
- download_third_parties_nix
- binary_linux_conda:
conda_docker_image: pytorch/conda-builder:cuda113
cuda_version: cu113
name: binary_linux_conda_py3.10_cu113
python_version: '3.10'
requires:
- download_third_parties_nix
- binary_linux_conda:
conda_docker_image: pytorch/conda-builder:cuda115
cuda_version: cu115
name: binary_linux_conda_py3.10_cu115
python_version: '3.10'
requires:
- download_third_parties_nix
- binary_macos_conda: - binary_macos_conda:
conda_docker_image: pytorch/conda-builder:cpu conda_docker_image: pytorch/conda-builder:cpu
cuda_version: cpu cuda_version: cpu
...@@ -1055,6 +1168,13 @@ workflows: ...@@ -1055,6 +1168,13 @@ workflows:
python_version: '3.9' python_version: '3.9'
requires: requires:
- download_third_parties_nix - download_third_parties_nix
- binary_macos_conda:
conda_docker_image: pytorch/conda-builder:cpu
cuda_version: cpu
name: binary_macos_conda_py3.10_cpu
python_version: '3.10'
requires:
- download_third_parties_nix
- binary_windows_conda: - binary_windows_conda:
conda_docker_image: pytorch/conda-builder:cpu conda_docker_image: pytorch/conda-builder:cpu
cuda_version: cpu cuda_version: cpu
...@@ -1100,6 +1220,21 @@ workflows: ...@@ -1100,6 +1220,21 @@ workflows:
cuda_version: cu115 cuda_version: cu115
name: binary_windows_conda_py3.9_cu115 name: binary_windows_conda_py3.9_cu115
python_version: '3.9' python_version: '3.9'
- binary_windows_conda:
conda_docker_image: pytorch/conda-builder:cpu
cuda_version: cpu
name: binary_windows_conda_py3.10_cpu
python_version: '3.10'
- binary_windows_conda:
conda_docker_image: pytorch/conda-builder:cuda113
cuda_version: cu113
name: binary_windows_conda_py3.10_cu113
python_version: '3.10'
- binary_windows_conda:
conda_docker_image: pytorch/conda-builder:cuda115
cuda_version: cu115
name: binary_windows_conda_py3.10_cu115
python_version: '3.10'
- build_docs: - build_docs:
filters: filters:
branches: branches:
...@@ -1154,6 +1289,12 @@ workflows: ...@@ -1154,6 +1289,12 @@ workflows:
python_version: '3.9' python_version: '3.9'
requires: requires:
- download_third_parties_nix - download_third_parties_nix
- unittest_linux_cpu:
cuda_version: cpu
name: unittest_linux_cpu_py3.10
python_version: '3.10'
requires:
- download_third_parties_nix
- unittest_linux_gpu: - unittest_linux_gpu:
cuda_version: cu113 cuda_version: cu113
name: unittest_linux_gpu_py3.7 name: unittest_linux_gpu_py3.7
...@@ -1172,6 +1313,12 @@ workflows: ...@@ -1172,6 +1313,12 @@ workflows:
python_version: '3.9' python_version: '3.9'
requires: requires:
- download_third_parties_nix - download_third_parties_nix
- unittest_linux_gpu:
cuda_version: cu113
name: unittest_linux_gpu_py3.10
python_version: '3.10'
requires:
- download_third_parties_nix
- unittest_windows_cpu: - unittest_windows_cpu:
cuda_version: cpu cuda_version: cpu
name: unittest_windows_cpu_py3.7 name: unittest_windows_cpu_py3.7
...@@ -1184,6 +1331,10 @@ workflows: ...@@ -1184,6 +1331,10 @@ workflows:
cuda_version: cpu cuda_version: cpu
name: unittest_windows_cpu_py3.9 name: unittest_windows_cpu_py3.9
python_version: '3.9' python_version: '3.9'
- unittest_windows_cpu:
cuda_version: cpu
name: unittest_windows_cpu_py3.10
python_version: '3.10'
- unittest_windows_gpu: - unittest_windows_gpu:
cuda_version: cu113 cuda_version: cu113
name: unittest_windows_gpu_py3.7 name: unittest_windows_gpu_py3.7
...@@ -1196,6 +1347,10 @@ workflows: ...@@ -1196,6 +1347,10 @@ workflows:
cuda_version: cu113 cuda_version: cu113
name: unittest_windows_gpu_py3.9 name: unittest_windows_gpu_py3.9
python_version: '3.9' python_version: '3.9'
- unittest_windows_gpu:
cuda_version: cu113
name: unittest_windows_gpu_py3.10
python_version: '3.10'
- unittest_macos_cpu: - unittest_macos_cpu:
cuda_version: cpu cuda_version: cpu
name: unittest_macos_cpu_py3.7 name: unittest_macos_cpu_py3.7
...@@ -1214,6 +1369,12 @@ workflows: ...@@ -1214,6 +1369,12 @@ workflows:
python_version: '3.9' python_version: '3.9'
requires: requires:
- download_third_parties_nix - download_third_parties_nix
- unittest_macos_cpu:
cuda_version: cpu
name: unittest_macos_cpu_py3.10
python_version: '3.10'
requires:
- download_third_parties_nix
nightly: nightly:
jobs: jobs:
- circleci_consistency: - circleci_consistency:
...@@ -2002,7 +2163,7 @@ workflows: ...@@ -2002,7 +2163,7 @@ workflows:
python_version: '3.9' python_version: '3.9'
requires: requires:
- nightly_binary_linux_wheel_py3.9_rocm4.5.2_upload - nightly_binary_linux_wheel_py3.9_rocm4.5.2_upload
- binary_macos_wheel: - binary_linux_wheel:
cuda_version: cpu cuda_version: cpu
filters: filters:
branches: branches:
...@@ -2010,8 +2171,8 @@ workflows: ...@@ -2010,8 +2171,8 @@ 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_macos_wheel_py3.7_cpu name: nightly_binary_linux_wheel_py3.10_cpu
python_version: '3.7' python_version: '3.10'
requires: requires:
- download_third_parties_nix - download_third_parties_nix
- binary_wheel_upload: - binary_wheel_upload:
...@@ -2022,11 +2183,11 @@ workflows: ...@@ -2022,11 +2183,11 @@ 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_macos_wheel_py3.7_cpu_upload name: nightly_binary_linux_wheel_py3.10_cpu_upload
requires: requires:
- nightly_binary_macos_wheel_py3.7_cpu - nightly_binary_linux_wheel_py3.10_cpu
subfolder: '' subfolder: cpu/
- binary_macos_wheel: - smoke_test_linux_pip:
cuda_version: cpu cuda_version: cpu
filters: filters:
branches: branches:
...@@ -2034,56 +2195,60 @@ workflows: ...@@ -2034,56 +2195,60 @@ 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_macos_wheel_py3.8_cpu name: nightly_binary_linux_wheel_py3.10_cpu_smoke_test_pip
python_version: '3.8' python_version: '3.10'
requires: requires:
- download_third_parties_nix - nightly_binary_linux_wheel_py3.10_cpu_upload
- binary_wheel_upload: - binary_linux_wheel:
context: org-member cuda_version: cu102
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_macos_wheel_py3.8_cpu_upload name: nightly_binary_linux_wheel_py3.10_cu102
python_version: '3.10'
requires: requires:
- nightly_binary_macos_wheel_py3.8_cpu - download_third_parties_nix
subfolder: '' wheel_docker_image: pytorch/manylinux-cuda102
- binary_macos_wheel: - binary_wheel_upload:
cuda_version: cpu context: org-member
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_macos_wheel_py3.9_cpu name: nightly_binary_linux_wheel_py3.10_cu102_upload
python_version: '3.9'
requires: requires:
- download_third_parties_nix - nightly_binary_linux_wheel_py3.10_cu102
- binary_wheel_upload: subfolder: cu102/
context: org-member - smoke_test_linux_pip:
cuda_version: cu102
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_macos_wheel_py3.9_cpu_upload name: nightly_binary_linux_wheel_py3.10_cu102_smoke_test_pip
python_version: '3.10'
requires: requires:
- nightly_binary_macos_wheel_py3.9_cpu - nightly_binary_linux_wheel_py3.10_cu102_upload
subfolder: '' - binary_linux_wheel:
- binary_windows_wheel: cuda_version: cu111
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_windows_wheel_py3.7_cpu name: nightly_binary_linux_wheel_py3.10_cu111
python_version: '3.7' python_version: '3.10'
requires:
- download_third_parties_nix
wheel_docker_image: pytorch/manylinux-cuda111
- binary_wheel_upload: - binary_wheel_upload:
context: org-member context: org-member
filters: filters:
...@@ -2092,23 +2257,23 @@ workflows: ...@@ -2092,23 +2257,23 @@ 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_windows_wheel_py3.7_cpu_upload name: nightly_binary_linux_wheel_py3.10_cu111_upload
requires: requires:
- nightly_binary_windows_wheel_py3.7_cpu - nightly_binary_linux_wheel_py3.10_cu111
subfolder: cpu/ subfolder: cu111/
- smoke_test_windows_pip: - smoke_test_linux_pip:
cuda_version: cpu cuda_version: cu111
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_windows_wheel_py3.7_cpu_smoke_test_pip name: nightly_binary_linux_wheel_py3.10_cu111_smoke_test_pip
python_version: '3.7' python_version: '3.10'
requires: requires:
- nightly_binary_windows_wheel_py3.7_cpu_upload - nightly_binary_linux_wheel_py3.10_cu111_upload
- binary_windows_wheel: - binary_linux_wheel:
cuda_version: cu113 cuda_version: cu113
filters: filters:
branches: branches:
...@@ -2116,8 +2281,10 @@ workflows: ...@@ -2116,8 +2281,10 @@ 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_windows_wheel_py3.7_cu113 name: nightly_binary_linux_wheel_py3.10_cu113
python_version: '3.7' python_version: '3.10'
requires:
- download_third_parties_nix
wheel_docker_image: pytorch/manylinux-cuda113 wheel_docker_image: pytorch/manylinux-cuda113
- binary_wheel_upload: - binary_wheel_upload:
context: org-member context: org-member
...@@ -2127,11 +2294,11 @@ workflows: ...@@ -2127,11 +2294,11 @@ 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_windows_wheel_py3.7_cu113_upload name: nightly_binary_linux_wheel_py3.10_cu113_upload
requires: requires:
- nightly_binary_windows_wheel_py3.7_cu113 - nightly_binary_linux_wheel_py3.10_cu113
subfolder: cu113/ subfolder: cu113/
- smoke_test_windows_pip: - smoke_test_linux_pip:
cuda_version: cu113 cuda_version: cu113
filters: filters:
branches: branches:
...@@ -2139,11 +2306,11 @@ workflows: ...@@ -2139,11 +2306,11 @@ 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_windows_wheel_py3.7_cu113_smoke_test_pip name: nightly_binary_linux_wheel_py3.10_cu113_smoke_test_pip
python_version: '3.7' python_version: '3.10'
requires: requires:
- nightly_binary_windows_wheel_py3.7_cu113_upload - nightly_binary_linux_wheel_py3.10_cu113_upload
- binary_windows_wheel: - binary_linux_wheel:
cuda_version: cu115 cuda_version: cu115
filters: filters:
branches: branches:
...@@ -2151,8 +2318,10 @@ workflows: ...@@ -2151,8 +2318,10 @@ 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_windows_wheel_py3.7_cu115 name: nightly_binary_linux_wheel_py3.10_cu115
python_version: '3.7' python_version: '3.10'
requires:
- download_third_parties_nix
wheel_docker_image: pytorch/manylinux-cuda115 wheel_docker_image: pytorch/manylinux-cuda115
- binary_wheel_upload: - binary_wheel_upload:
context: org-member context: org-member
...@@ -2162,11 +2331,11 @@ workflows: ...@@ -2162,11 +2331,11 @@ 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_windows_wheel_py3.7_cu115_upload name: nightly_binary_linux_wheel_py3.10_cu115_upload
requires: requires:
- nightly_binary_windows_wheel_py3.7_cu115 - nightly_binary_linux_wheel_py3.10_cu115
subfolder: cu115/ subfolder: cu115/
- smoke_test_windows_pip: - smoke_test_linux_pip:
cuda_version: cu115 cuda_version: cu115
filters: filters:
branches: branches:
...@@ -2174,20 +2343,23 @@ workflows: ...@@ -2174,20 +2343,23 @@ 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_windows_wheel_py3.7_cu115_smoke_test_pip name: nightly_binary_linux_wheel_py3.10_cu115_smoke_test_pip
python_version: '3.7' python_version: '3.10'
requires: requires:
- nightly_binary_windows_wheel_py3.7_cu115_upload - nightly_binary_linux_wheel_py3.10_cu115_upload
- binary_windows_wheel: - binary_linux_wheel:
cuda_version: cpu 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_windows_wheel_py3.8_cpu name: nightly_binary_linux_wheel_py3.10_rocm4.3.1
python_version: '3.8' python_version: '3.10'
requires:
- download_third_parties_nix
wheel_docker_image: pytorch/manylinux-rocm:4.3.1
- binary_wheel_upload: - binary_wheel_upload:
context: org-member context: org-member
filters: filters:
...@@ -2196,33 +2368,35 @@ workflows: ...@@ -2196,33 +2368,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_windows_wheel_py3.8_cpu_upload name: nightly_binary_linux_wheel_py3.10_rocm4.3.1_upload
requires: requires:
- nightly_binary_windows_wheel_py3.8_cpu - nightly_binary_linux_wheel_py3.10_rocm4.3.1
subfolder: cpu/ subfolder: rocm4.3.1/
- smoke_test_windows_pip: - smoke_test_linux_pip:
cuda_version: cpu 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_windows_wheel_py3.8_cpu_smoke_test_pip name: nightly_binary_linux_wheel_py3.10_rocm4.3.1_smoke_test_pip
python_version: '3.8' python_version: '3.10'
requires: requires:
- nightly_binary_windows_wheel_py3.8_cpu_upload - nightly_binary_linux_wheel_py3.10_rocm4.3.1_upload
- binary_windows_wheel: - binary_linux_wheel:
cuda_version: cu113 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_windows_wheel_py3.8_cu113 name: nightly_binary_linux_wheel_py3.10_rocm4.5.2
python_version: '3.8' python_version: '3.10'
wheel_docker_image: pytorch/manylinux-cuda113 requires:
- 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:
...@@ -2231,33 +2405,34 @@ workflows: ...@@ -2231,33 +2405,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_windows_wheel_py3.8_cu113_upload name: nightly_binary_linux_wheel_py3.10_rocm4.5.2_upload
requires: requires:
- nightly_binary_windows_wheel_py3.8_cu113 - nightly_binary_linux_wheel_py3.10_rocm4.5.2
subfolder: cu113/ subfolder: rocm4.5.2/
- smoke_test_windows_pip: - smoke_test_linux_pip:
cuda_version: cu113 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_windows_wheel_py3.8_cu113_smoke_test_pip name: nightly_binary_linux_wheel_py3.10_rocm4.5.2_smoke_test_pip
python_version: '3.8' python_version: '3.10'
requires: requires:
- nightly_binary_windows_wheel_py3.8_cu113_upload - nightly_binary_linux_wheel_py3.10_rocm4.5.2_upload
- binary_windows_wheel: - binary_macos_wheel:
cuda_version: cu115 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_windows_wheel_py3.8_cu115 name: nightly_binary_macos_wheel_py3.7_cpu
python_version: '3.8' python_version: '3.7'
wheel_docker_image: pytorch/manylinux-cuda115 requires:
- download_third_parties_nix
- binary_wheel_upload: - binary_wheel_upload:
context: org-member context: org-member
filters: filters:
...@@ -2266,24 +2441,292 @@ workflows: ...@@ -2266,24 +2441,292 @@ 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_windows_wheel_py3.8_cu115_upload name: nightly_binary_macos_wheel_py3.7_cpu_upload
requires: requires:
- nightly_binary_windows_wheel_py3.8_cu115 - nightly_binary_macos_wheel_py3.7_cpu
subfolder: cu115/ subfolder: ''
- smoke_test_windows_pip: - binary_macos_wheel:
cuda_version: cu115 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_windows_wheel_py3.8_cu115_smoke_test_pip name: nightly_binary_macos_wheel_py3.8_cpu
python_version: '3.8' python_version: '3.8'
requires: requires:
- nightly_binary_windows_wheel_py3.8_cu115_upload - download_third_parties_nix
- binary_windows_wheel: - binary_wheel_upload:
cuda_version: cpu context: org-member
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_macos_wheel_py3.8_cpu_upload
requires:
- nightly_binary_macos_wheel_py3.8_cpu
subfolder: ''
- binary_macos_wheel:
cuda_version: cpu
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_macos_wheel_py3.9_cpu
python_version: '3.9'
requires:
- download_third_parties_nix
- binary_wheel_upload:
context: org-member
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_macos_wheel_py3.9_cpu_upload
requires:
- nightly_binary_macos_wheel_py3.9_cpu
subfolder: ''
- binary_macos_wheel:
cuda_version: cpu
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_macos_wheel_py3.10_cpu
python_version: '3.10'
requires:
- download_third_parties_nix
- binary_wheel_upload:
context: org-member
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_macos_wheel_py3.10_cpu_upload
requires:
- nightly_binary_macos_wheel_py3.10_cpu
subfolder: ''
- binary_windows_wheel:
cuda_version: cpu
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_windows_wheel_py3.7_cpu
python_version: '3.7'
- binary_wheel_upload:
context: org-member
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_windows_wheel_py3.7_cpu_upload
requires:
- nightly_binary_windows_wheel_py3.7_cpu
subfolder: cpu/
- smoke_test_windows_pip:
cuda_version: cpu
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_windows_wheel_py3.7_cpu_smoke_test_pip
python_version: '3.7'
requires:
- nightly_binary_windows_wheel_py3.7_cpu_upload
- binary_windows_wheel:
cuda_version: cu113
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_windows_wheel_py3.7_cu113
python_version: '3.7'
wheel_docker_image: pytorch/manylinux-cuda113
- binary_wheel_upload:
context: org-member
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_windows_wheel_py3.7_cu113_upload
requires:
- nightly_binary_windows_wheel_py3.7_cu113
subfolder: cu113/
- smoke_test_windows_pip:
cuda_version: cu113
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_windows_wheel_py3.7_cu113_smoke_test_pip
python_version: '3.7'
requires:
- nightly_binary_windows_wheel_py3.7_cu113_upload
- binary_windows_wheel:
cuda_version: cu115
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_windows_wheel_py3.7_cu115
python_version: '3.7'
wheel_docker_image: pytorch/manylinux-cuda115
- binary_wheel_upload:
context: org-member
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_windows_wheel_py3.7_cu115_upload
requires:
- nightly_binary_windows_wheel_py3.7_cu115
subfolder: cu115/
- smoke_test_windows_pip:
cuda_version: cu115
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_windows_wheel_py3.7_cu115_smoke_test_pip
python_version: '3.7'
requires:
- nightly_binary_windows_wheel_py3.7_cu115_upload
- binary_windows_wheel:
cuda_version: cpu
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_windows_wheel_py3.8_cpu
python_version: '3.8'
- binary_wheel_upload:
context: org-member
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_windows_wheel_py3.8_cpu_upload
requires:
- nightly_binary_windows_wheel_py3.8_cpu
subfolder: cpu/
- smoke_test_windows_pip:
cuda_version: cpu
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_windows_wheel_py3.8_cpu_smoke_test_pip
python_version: '3.8'
requires:
- nightly_binary_windows_wheel_py3.8_cpu_upload
- binary_windows_wheel:
cuda_version: cu113
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_windows_wheel_py3.8_cu113
python_version: '3.8'
wheel_docker_image: pytorch/manylinux-cuda113
- binary_wheel_upload:
context: org-member
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_windows_wheel_py3.8_cu113_upload
requires:
- nightly_binary_windows_wheel_py3.8_cu113
subfolder: cu113/
- smoke_test_windows_pip:
cuda_version: cu113
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_windows_wheel_py3.8_cu113_smoke_test_pip
python_version: '3.8'
requires:
- nightly_binary_windows_wheel_py3.8_cu113_upload
- binary_windows_wheel:
cuda_version: cu115
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_windows_wheel_py3.8_cu115
python_version: '3.8'
wheel_docker_image: pytorch/manylinux-cuda115
- binary_wheel_upload:
context: org-member
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_windows_wheel_py3.8_cu115_upload
requires:
- nightly_binary_windows_wheel_py3.8_cu115
subfolder: cu115/
- smoke_test_windows_pip:
cuda_version: cu115
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_windows_wheel_py3.8_cu115_smoke_test_pip
python_version: '3.8'
requires:
- nightly_binary_windows_wheel_py3.8_cu115_upload
- binary_windows_wheel:
cuda_version: cpu
filters: filters:
branches: branches:
only: only:
...@@ -2347,10 +2790,114 @@ workflows: ...@@ -2347,10 +2790,114 @@ 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_windows_wheel_py3.9_cu113_smoke_test_pip name: nightly_binary_windows_wheel_py3.9_cu113_smoke_test_pip
python_version: '3.9' python_version: '3.9'
requires:
- nightly_binary_windows_wheel_py3.9_cu113_upload
- binary_windows_wheel:
cuda_version: cu115
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_windows_wheel_py3.9_cu115
python_version: '3.9'
wheel_docker_image: pytorch/manylinux-cuda115
- binary_wheel_upload:
context: org-member
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_windows_wheel_py3.9_cu115_upload
requires:
- nightly_binary_windows_wheel_py3.9_cu115
subfolder: cu115/
- smoke_test_windows_pip:
cuda_version: cu115
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_windows_wheel_py3.9_cu115_smoke_test_pip
python_version: '3.9'
requires:
- nightly_binary_windows_wheel_py3.9_cu115_upload
- binary_windows_wheel:
cuda_version: cpu
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_windows_wheel_py3.10_cpu
python_version: '3.10'
- binary_wheel_upload:
context: org-member
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_windows_wheel_py3.10_cpu_upload
requires:
- nightly_binary_windows_wheel_py3.10_cpu
subfolder: cpu/
- smoke_test_windows_pip:
cuda_version: cpu
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_windows_wheel_py3.10_cpu_smoke_test_pip
python_version: '3.10'
requires:
- nightly_binary_windows_wheel_py3.10_cpu_upload
- binary_windows_wheel:
cuda_version: cu113
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_windows_wheel_py3.10_cu113
python_version: '3.10'
wheel_docker_image: pytorch/manylinux-cuda113
- binary_wheel_upload:
context: org-member
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_windows_wheel_py3.10_cu113_upload
requires:
- nightly_binary_windows_wheel_py3.10_cu113
subfolder: cu113/
- smoke_test_windows_pip:
cuda_version: cu113
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_windows_wheel_py3.10_cu113_smoke_test_pip
python_version: '3.10'
requires: requires:
- nightly_binary_windows_wheel_py3.9_cu113_upload - nightly_binary_windows_wheel_py3.10_cu113_upload
- binary_windows_wheel: - binary_windows_wheel:
cuda_version: cu115 cuda_version: cu115
filters: filters:
...@@ -2359,8 +2906,8 @@ workflows: ...@@ -2359,8 +2906,8 @@ 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_windows_wheel_py3.9_cu115 name: nightly_binary_windows_wheel_py3.10_cu115
python_version: '3.9' python_version: '3.10'
wheel_docker_image: pytorch/manylinux-cuda115 wheel_docker_image: pytorch/manylinux-cuda115
- binary_wheel_upload: - binary_wheel_upload:
context: org-member context: org-member
...@@ -2370,9 +2917,9 @@ workflows: ...@@ -2370,9 +2917,9 @@ 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_windows_wheel_py3.9_cu115_upload name: nightly_binary_windows_wheel_py3.10_cu115_upload
requires: requires:
- nightly_binary_windows_wheel_py3.9_cu115 - nightly_binary_windows_wheel_py3.10_cu115
subfolder: cu115/ subfolder: cu115/
- smoke_test_windows_pip: - smoke_test_windows_pip:
cuda_version: cu115 cuda_version: cu115
...@@ -2382,10 +2929,10 @@ workflows: ...@@ -2382,10 +2929,10 @@ 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_windows_wheel_py3.9_cu115_smoke_test_pip name: nightly_binary_windows_wheel_py3.10_cu115_smoke_test_pip
python_version: '3.9' python_version: '3.10'
requires: requires:
- nightly_binary_windows_wheel_py3.9_cu115_upload - nightly_binary_windows_wheel_py3.10_cu115_upload
- binary_linux_conda: - binary_linux_conda:
conda_docker_image: pytorch/conda-builder:cpu conda_docker_image: pytorch/conda-builder:cpu
cuda_version: cpu cuda_version: cpu
...@@ -2926,6 +3473,186 @@ workflows: ...@@ -2926,6 +3473,186 @@ workflows:
python_version: '3.9' python_version: '3.9'
requires: requires:
- nightly_binary_linux_conda_py3.9_cu115_upload - nightly_binary_linux_conda_py3.9_cu115_upload
- binary_linux_conda:
conda_docker_image: pytorch/conda-builder:cpu
cuda_version: cpu
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_conda_py3.10_cpu
python_version: '3.10'
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.10_cpu_upload
requires:
- nightly_binary_linux_conda_py3.10_cpu
- smoke_test_linux_conda:
cuda_version: cpu
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_conda_py3.10_cpu_smoke_test_conda
python_version: '3.10'
requires:
- nightly_binary_linux_conda_py3.10_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.10_cu102
python_version: '3.10'
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.10_cu102_upload
requires:
- nightly_binary_linux_conda_py3.10_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.10_cu102_smoke_test_conda
python_version: '3.10'
requires:
- nightly_binary_linux_conda_py3.10_cu102_upload
- binary_linux_conda:
conda_docker_image: pytorch/conda-builder:cuda111
cuda_version: cu111
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_conda_py3.10_cu111
python_version: '3.10'
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.10_cu111_upload
requires:
- nightly_binary_linux_conda_py3.10_cu111
- smoke_test_linux_conda_gpu:
cuda_version: cu111
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_conda_py3.10_cu111_smoke_test_conda
python_version: '3.10'
requires:
- nightly_binary_linux_conda_py3.10_cu111_upload
- binary_linux_conda:
conda_docker_image: pytorch/conda-builder:cuda113
cuda_version: cu113
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_conda_py3.10_cu113
python_version: '3.10'
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.10_cu113_upload
requires:
- nightly_binary_linux_conda_py3.10_cu113
- smoke_test_linux_conda_gpu:
cuda_version: cu113
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_conda_py3.10_cu113_smoke_test_conda
python_version: '3.10'
requires:
- nightly_binary_linux_conda_py3.10_cu113_upload
- binary_linux_conda:
conda_docker_image: pytorch/conda-builder:cuda115
cuda_version: cu115
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_conda_py3.10_cu115
python_version: '3.10'
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.10_cu115_upload
requires:
- nightly_binary_linux_conda_py3.10_cu115
- smoke_test_linux_conda_gpu:
cuda_version: cu115
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_conda_py3.10_cu115_smoke_test_conda
python_version: '3.10'
requires:
- nightly_binary_linux_conda_py3.10_cu115_upload
- binary_macos_conda: - binary_macos_conda:
conda_docker_image: pytorch/conda-builder:cpu conda_docker_image: pytorch/conda-builder:cpu
cuda_version: cpu cuda_version: cpu
...@@ -2998,6 +3725,30 @@ workflows: ...@@ -2998,6 +3725,30 @@ workflows:
name: nightly_binary_macos_conda_py3.9_cpu_upload name: nightly_binary_macos_conda_py3.9_cpu_upload
requires: requires:
- nightly_binary_macos_conda_py3.9_cpu - nightly_binary_macos_conda_py3.9_cpu
- binary_macos_conda:
conda_docker_image: pytorch/conda-builder:cpu
cuda_version: cpu
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_macos_conda_py3.10_cpu
python_version: '3.10'
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_macos_conda_py3.10_cpu_upload
requires:
- nightly_binary_macos_conda_py3.10_cpu
- binary_windows_conda: - binary_windows_conda:
conda_docker_image: pytorch/conda-builder:cpu conda_docker_image: pytorch/conda-builder:cpu
cuda_version: cpu cuda_version: cpu
...@@ -3304,3 +4055,105 @@ workflows: ...@@ -3304,3 +4055,105 @@ workflows:
python_version: '3.9' python_version: '3.9'
requires: requires:
- nightly_binary_windows_conda_py3.9_cu115_upload - nightly_binary_windows_conda_py3.9_cu115_upload
- binary_windows_conda:
conda_docker_image: pytorch/conda-builder:cpu
cuda_version: cpu
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_windows_conda_py3.10_cpu
python_version: '3.10'
- binary_conda_upload:
context: org-member
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_windows_conda_py3.10_cpu_upload
requires:
- nightly_binary_windows_conda_py3.10_cpu
- smoke_test_windows_conda:
cuda_version: cpu
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_windows_conda_py3.10_cpu_smoke_test_conda
python_version: '3.10'
requires:
- nightly_binary_windows_conda_py3.10_cpu_upload
- binary_windows_conda:
conda_docker_image: pytorch/conda-builder:cuda113
cuda_version: cu113
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_windows_conda_py3.10_cu113
python_version: '3.10'
- binary_conda_upload:
context: org-member
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_windows_conda_py3.10_cu113_upload
requires:
- nightly_binary_windows_conda_py3.10_cu113
- smoke_test_windows_conda:
cuda_version: cu113
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_windows_conda_py3.10_cu113_smoke_test_conda
python_version: '3.10'
requires:
- nightly_binary_windows_conda_py3.10_cu113_upload
- binary_windows_conda:
conda_docker_image: pytorch/conda-builder:cuda115
cuda_version: cu115
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_windows_conda_py3.10_cu115
python_version: '3.10'
- binary_conda_upload:
context: org-member
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_windows_conda_py3.10_cu115_upload
requires:
- nightly_binary_windows_conda_py3.10_cu115
- smoke_test_windows_conda:
cuda_version: cu115
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_windows_conda_py3.10_cu115_smoke_test_conda
python_version: '3.10'
requires:
- nightly_binary_windows_conda_py3.10_cu115_upload
.circleci/config.yml.in
View file @ 27dff6ba
...@@ -475,6 +475,7 @@ jobs: ...@@ -475,6 +475,7 @@ jobs:
command: .circleci/unittest/linux/scripts/run_test.sh command: .circleci/unittest/linux/scripts/run_test.sh
environment: environment:
TORCHAUDIO_TEST_ALLOW_SKIP_IF_NO_CUDA: true TORCHAUDIO_TEST_ALLOW_SKIP_IF_NO_CUDA: true
TORCHAUDIO_TEST_ALLOW_SKIP_IF_ON_PYTHON_310: true
- store_test_results: - store_test_results:
path: test-results path: test-results
- store_artifacts: - store_artifacts:
...@@ -504,7 +505,10 @@ jobs: ...@@ -504,7 +505,10 @@ jobs:
command: docker run -t --gpus all -e UPLOAD_CHANNEL -e CONDA_CHANNEL_FLAGS -e BUILD_FFMPEG=1 -v $PWD:$PWD -w $PWD "${image_name}" .circleci/unittest/linux/scripts/install.sh command: docker run -t --gpus all -e UPLOAD_CHANNEL -e CONDA_CHANNEL_FLAGS -e BUILD_FFMPEG=1 -v $PWD:$PWD -w $PWD "${image_name}" .circleci/unittest/linux/scripts/install.sh
- run: - run:
name: Run tests name: Run tests
command: docker run -t --gpus all -v $PWD:$PWD -w $PWD -e "CI=${CI}" "${image_name}" .circleci/unittest/linux/scripts/run_test.sh environment:
TORCHAUDIO_TEST_ALLOW_SKIP_IF_ON_PYTHON_310: true
command: |
docker run -t --gpus all -v $PWD:$PWD -w $PWD -e "CI=${CI}" -e TORCHAUDIO_TEST_ALLOW_SKIP_IF_ON_PYTHON_310 "${image_name}" .circleci/unittest/linux/scripts/run_test.sh
- store_test_results: - store_test_results:
path: test-results path: test-results
- store_artifacts: - store_artifacts:
...@@ -540,6 +544,8 @@ jobs: ...@@ -540,6 +544,8 @@ jobs:
TORCHAUDIO_TEST_ALLOW_SKIP_IF_NO_CUDA: true TORCHAUDIO_TEST_ALLOW_SKIP_IF_NO_CUDA: true
TORCHAUDIO_TEST_ALLOW_SKIP_IF_NO_KALDI: true TORCHAUDIO_TEST_ALLOW_SKIP_IF_NO_KALDI: true
TORCHAUDIO_TEST_ALLOW_SKIP_IF_NO_SOX: true TORCHAUDIO_TEST_ALLOW_SKIP_IF_NO_SOX: true
TORCHAUDIO_TEST_ALLOW_SKIP_IF_ON_PYTHON_310: true
TORCHAUDIO_TEST_ALLOW_SKIP_IF_NO_MOD_sentencepiece: true
- store_test_results: - store_test_results:
path: test-results path: test-results
- store_artifacts: - store_artifacts:
...@@ -583,6 +589,8 @@ jobs: ...@@ -583,6 +589,8 @@ jobs:
TORCHAUDIO_TEST_ALLOW_SKIP_IF_NO_CTC_DECODER: true TORCHAUDIO_TEST_ALLOW_SKIP_IF_NO_CTC_DECODER: true
TORCHAUDIO_TEST_ALLOW_SKIP_IF_NO_KALDI: true TORCHAUDIO_TEST_ALLOW_SKIP_IF_NO_KALDI: true
TORCHAUDIO_TEST_ALLOW_SKIP_IF_NO_SOX: true TORCHAUDIO_TEST_ALLOW_SKIP_IF_NO_SOX: true
TORCHAUDIO_TEST_ALLOW_SKIP_IF_ON_PYTHON_310: true
TORCHAUDIO_TEST_ALLOW_SKIP_IF_NO_MOD_sentencepiece: true
- store_test_results: - store_test_results:
path: test-results path: test-results
- store_artifacts: - store_artifacts:
...@@ -618,6 +626,8 @@ jobs: ...@@ -618,6 +626,8 @@ jobs:
TORCHAUDIO_TEST_ALLOW_SKIP_IF_NO_CMD_COMPUTE_SPECTROGRAM_FEATS: true TORCHAUDIO_TEST_ALLOW_SKIP_IF_NO_CMD_COMPUTE_SPECTROGRAM_FEATS: true
TORCHAUDIO_TEST_ALLOW_SKIP_IF_NO_CUDA: true TORCHAUDIO_TEST_ALLOW_SKIP_IF_NO_CUDA: true
TORCHAUDIO_TEST_ALLOW_SKIP_IF_NO_QUANTIZATION: true TORCHAUDIO_TEST_ALLOW_SKIP_IF_NO_QUANTIZATION: true
TORCHAUDIO_TEST_ALLOW_SKIP_IF_ON_PYTHON_310: true
TORCHAUDIO_TEST_ALLOW_SKIP_IF_NO_MOD_sentencepiece: true
- store_test_results: - store_test_results:
path: test-results path: test-results
- store_artifacts: - store_artifacts:
......
.circleci/regenerate.py
View file @ 27dff6ba
...@@ -21,7 +21,7 @@ import yaml ...@@ -21,7 +21,7 @@ import yaml
from jinja2 import select_autoescape from jinja2 import select_autoescape
PYTHON_VERSIONS = ["3.7", "3.8", "3.9"] PYTHON_VERSIONS = ["3.7", "3.8", "3.9", "3.10"]
CU_VERSIONS_DICT = { CU_VERSIONS_DICT = {
"linux": ["cpu", "cu102", "cu111", "cu113", "cu115", "rocm4.3.1", "rocm4.5.2"], "linux": ["cpu", "cu102", "cu111", "cu113", "cu115", "rocm4.3.1", "rocm4.5.2"],
"windows": ["cpu", "cu113", "cu115"], "windows": ["cpu", "cu113", "cu115"],
......
.circleci/unittest/linux/scripts/install.sh
View file @ 27dff6ba
...@@ -37,6 +37,10 @@ fi ...@@ -37,6 +37,10 @@ fi
printf "Installing PyTorch with %s\n" "${cudatoolkit}" printf "Installing PyTorch with %s\n" "${cudatoolkit}"
( (
if [[ "$(python --version)" = *3.10* ]]; then
CONDA_CHANNEL_FLAGS="-c conda-forge"
fi
if [ "${os}" == MacOSX ] ; then if [ "${os}" == MacOSX ] ; then
# TODO: this can be removed as soon as linking issue could be resolved # TODO: this can be removed as soon as linking issue could be resolved
# see https://github.com/pytorch/pytorch/issues/62424 from details # see https://github.com/pytorch/pytorch/issues/62424 from details
...@@ -57,8 +61,8 @@ python setup.py install ...@@ -57,8 +61,8 @@ python setup.py install
# 3. Install Test tools # 3. Install Test tools
printf "* Installing test tools\n" printf "* Installing test tools\n"
NUMBA_DEV_CHANNEL="" NUMBA_DEV_CHANNEL=""
if [[ "$(python --version)" = *3.9* ]]; then if [[ "$(python --version)" = *3.9* || "$(python --version)" = *3.10* ]]; then
# Numba isn't available for Python 3.9 except on the numba dev channel and building from source fails # Numba isn't available for Python 3.9 and 3.10 except on the numba dev channel and building from source fails
# See https://github.com/librosa/librosa/issues/1270#issuecomment-759065048 # See https://github.com/librosa/librosa/issues/1270#issuecomment-759065048
NUMBA_DEV_CHANNEL="-c numba/label/dev" NUMBA_DEV_CHANNEL="-c numba/label/dev"
fi fi
......
.circleci/unittest/linux/scripts/setup_env.sh
View file @ 27dff6ba
...@@ -25,7 +25,15 @@ if [ ! -d "${conda_dir}" ]; then ...@@ -25,7 +25,15 @@ if [ ! -d "${conda_dir}" ]; then
bash ./miniconda.sh -b -f -p "${conda_dir}" bash ./miniconda.sh -b -f -p "${conda_dir}"
eval "$("${conda_dir}/bin/conda" shell.bash hook)" eval "$("${conda_dir}/bin/conda" shell.bash hook)"
printf "* Updating the base Python version to %s\n" "${PYTHON_VERSION}" printf "* Updating the base Python version to %s\n" "${PYTHON_VERSION}"
conda install --quiet -y python="${PYTHON_VERSION}"
ADDITIONAL_CHANNELS=""
if [[ ${PYTHON_VERSION} == 3.10 ]]; then
ADDITIONAL_CHANNELS="-c conda-forge"
fi
# Need to disable shell check since this'll fail out if ADDITIONAL_CHANNELS is empty
# shellcheck disable=SC2086
conda install ${ADDITIONAL_CHANNELS} --quiet -y python="${PYTHON_VERSION}"
else else
eval "$("${conda_dir}/bin/conda" shell.bash hook)" eval "$("${conda_dir}/bin/conda" shell.bash hook)"
fi fi
......
.circleci/unittest/windows/scripts/install.sh
View file @ 27dff6ba
...@@ -48,16 +48,39 @@ printf "* Installing torchaudio\n" ...@@ -48,16 +48,39 @@ printf "* Installing torchaudio\n"
# 3. Install Test tools # 3. Install Test tools
printf "* Installing test tools\n" printf "* Installing test tools\n"
NUMBA_DEV_CHANNEL="" NUMBA_DEV_CHANNEL=""
if [[ "$(python --version)" = *3.9* ]]; then SENTENCEPIECE_DEPENDENCY="sentencepiece"
# Numba isn't available for Python 3.9 except on the numba dev channel and building from source fails case "$(python --version)" in
# See https://github.com/librosa/librosa/issues/1270#issuecomment-759065048 *3.9*)
NUMBA_DEV_CHANNEL="-c numba/label/dev" # Numba isn't available for Python 3.9 except on the numba dev channel and building from source fails
fi # See https://github.com/librosa/librosa/issues/1270#issuecomment-759065048
NUMBA_DEV_CHANNEL="-c numba/label/dev"
;;
*3.10*)
# Don't install sentencepiece, no python 3.10 dependencies available for windows yet
SENTENCEPIECE_DEPENDENCY=""
NUMBA_DEV_CHANNEL="-c numba/label/dev"
;;
esac
# Note: installing librosa via pip fail because it will try to compile numba. # Note: installing librosa via pip fail because it will try to compile numba.
( (
set -x set -x
conda install -y -c conda-forge ${NUMBA_DEV_CHANNEL} 'librosa>=0.8.0' parameterized 'requests>=2.20' conda install -y -c conda-forge ${NUMBA_DEV_CHANNEL} 'librosa>=0.8.0' parameterized 'requests>=2.20'
pip install kaldi-io SoundFile coverage pytest pytest-cov scipy transformers expecttest unidecode inflect Pillow sentencepiece pytorch-lightning # Need to disable shell check since this'll fail out if SENTENCEPIECE_DEPENDENCY is empty
# shellcheck disable=SC2086
pip install \
${SENTENCEPIECE_DEPENDENCY} \
Pillow \
SoundFile \
coverage \
expecttest \
inflect \
kaldi-io \
pytest \
pytest-cov \
pytorch-lightning \
scipy \
transformers \
unidecode
) )
# Install fairseq # Install fairseq
git clone https://github.com/pytorch/fairseq git clone https://github.com/pytorch/fairseq
......
.circleci/unittest/windows/scripts/setup_env.sh
View file @ 27dff6ba
...@@ -25,7 +25,15 @@ if [ ! -d "${conda_dir}" ]; then ...@@ -25,7 +25,15 @@ if [ ! -d "${conda_dir}" ]; then
unset miniconda_exe unset miniconda_exe
eval "$("${conda_dir}/Scripts/conda.exe" 'shell.bash' 'hook')" eval "$("${conda_dir}/Scripts/conda.exe" 'shell.bash' 'hook')"
printf "* Updating the base Python version to %s\n" "${PYTHON_VERSION}" printf "* Updating the base Python version to %s\n" "${PYTHON_VERSION}"
conda install --quiet -y python="$PYTHON_VERSION"
ADDITIONAL_CHANNELS=""
if [[ ${PYTHON_VERSION} == 3.10 ]]; then
ADDITIONAL_CHANNELS="-c conda-forge"
fi
# Need to disable shell check since this'll fail out if ADDITIONAL_CHANNELS is empty
# shellcheck disable=SC2086
conda install ${ADDITIONAL_CHANNELS} --quiet -y python="$PYTHON_VERSION"
else else
eval "$("${conda_dir}/Scripts/conda.exe" 'shell.bash' 'hook')" eval "$("${conda_dir}/Scripts/conda.exe" 'shell.bash' 'hook')"
fi fi
......
packaging/build_conda.sh
View file @ 27dff6ba
...@@ -12,9 +12,15 @@ setup_conda_cudatoolkit_constraint ...@@ -12,9 +12,15 @@ setup_conda_cudatoolkit_constraint
setup_visual_studio_constraint setup_visual_studio_constraint
# nvidia channel included for cudatoolkit >= 11 however for 11.5 we use conda-forge # nvidia channel included for cudatoolkit >= 11 however for 11.5 we use conda-forge
# HACK HACK HACK: Remove PYTHON_VERSION check once https://github.com/pytorch/builder/pull/961 is merged
export CUDATOOLKIT_CHANNEL="nvidia" export CUDATOOLKIT_CHANNEL="nvidia"
if [[ "$CU_VERSION" == cu115 ]]; then # NOTE: This is needed because `cudatoolkit=11.5` has a dependency on conda-forge
export CUDATOOLKIT_CHANNEL="conda-forge" # See: https://github.com/pytorch/audio/pull/2224#issuecomment-1049185550
if [[ ${CU_VERSION} = "cu115" ]]; then
CONDA_CHANNEL_FLAGS="${CONDA_CHANNEL_FLAGS} -c conda-forge"
fi fi
# NOTE: There are some dependencies that are not available for macOS on Python 3.10 without conda-forge
conda build -c defaults -c $CUDATOOLKIT_CHANNEL $CONDA_CHANNEL_FLAGS --no-anaconda-upload --python "$PYTHON_VERSION" packaging/torchaudio if [[ ${OSTYPE} =~ darwin* ]] && [[ ${PYTHON_VERSION} = "3.10" ]]; then
CONDA_CHANNEL_FLAGS="${CONDA_CHANNEL_FLAGS} -c conda-forge"
fi
conda build -c defaults -c $CUDATOOLKIT_CHANNEL ${CONDA_CHANNEL_FLAGS:-} --no-anaconda-upload --python "$PYTHON_VERSION" packaging/torchaudio
packaging/pkg_helpers.bash
View file @ 27dff6ba
...@@ -197,6 +197,7 @@ setup_wheel_python() { ...@@ -197,6 +197,7 @@ setup_wheel_python() {
3.7) python_abi=cp37-cp37m ;; 3.7) python_abi=cp37-cp37m ;;
3.8) python_abi=cp38-cp38 ;; 3.8) python_abi=cp38-cp38 ;;
3.9) python_abi=cp39-cp39 ;; 3.9) python_abi=cp39-cp39 ;;
3.10) python_abi=cp310-cp310 ;;
*) *)
echo "Unrecognized PYTHON_VERSION=$PYTHON_VERSION" echo "Unrecognized PYTHON_VERSION=$PYTHON_VERSION"
exit 1 exit 1
......
packaging/torchaudio/meta.yaml
View file @ 27dff6ba
...@@ -17,7 +17,8 @@ requirements: ...@@ -17,7 +17,8 @@ requirements:
- pkg-config # [not win] - pkg-config # [not win]
- cmake - cmake
- ninja - ninja
- defaults::numpy >=1.11 - numpy>=1.11 # [py <= 39]
- numpy>=1.21.2 # [py >= 310]
- pytorch-mutex 1.0 {{ build_variant }} # [not osx ] - pytorch-mutex 1.0 {{ build_variant }} # [not osx ]
{{ environ.get('CONDA_PYTORCH_BUILD_CONSTRAINT', 'pytorch') }} {{ environ.get('CONDA_PYTORCH_BUILD_CONSTRAINT', 'pytorch') }}
{{ environ.get('CONDA_EXTRA_BUILD_CONSTRAINT', '') }} {{ environ.get('CONDA_EXTRA_BUILD_CONSTRAINT', '') }}
...@@ -26,7 +27,8 @@ requirements: ...@@ -26,7 +27,8 @@ requirements:
run: run:
- python - python
- defaults::numpy >=1.11 - numpy>=1.11 # [py <= 39]
- numpy>=1.21.2 # [py >= 310]
- pytorch-mutex 1.0 {{ build_variant }} # [not osx ] - pytorch-mutex 1.0 {{ build_variant }} # [not osx ]
{{ environ.get('CONDA_PYTORCH_CONSTRAINT', 'pytorch') }} {{ environ.get('CONDA_PYTORCH_CONSTRAINT', 'pytorch') }}
{{ environ.get('CONDA_CUDATOOLKIT_CONSTRAINT', '') }} {{ environ.get('CONDA_CUDATOOLKIT_CONSTRAINT', '') }}
......
test/torchaudio_unittest/common_utils/__init__.py
View file @ 27dff6ba
...@@ -17,6 +17,7 @@ from .case_utils import ( ...@@ -17,6 +17,7 @@ from .case_utils import (
skipIfRocm, skipIfRocm,
skipIfNoQengine, skipIfNoQengine,
skipIfNoFFmpeg, skipIfNoFFmpeg,
skipIfPy310,
) )
from .data_utils import ( from .data_utils import (
get_asset_path, get_asset_path,
...@@ -59,6 +60,7 @@ __all__ = [ ...@@ -59,6 +60,7 @@ __all__ = [
"skipIfRocm", "skipIfRocm",
"skipIfNoQengine", "skipIfNoQengine",
"skipIfNoFFmpeg", "skipIfNoFFmpeg",
"skipIfPy310",
"get_wav_data", "get_wav_data",
"normalize_wav", "normalize_wav",
"load_wav", "load_wav",
......
test/torchaudio_unittest/common_utils/case_utils.py
View file @ 27dff6ba
...@@ -2,6 +2,7 @@ import functools ...@@ -2,6 +2,7 @@ import functools
import os.path import os.path
import shutil import shutil
import subprocess import subprocess
import sys
import tempfile import tempfile
import time import time
import unittest import unittest
...@@ -212,3 +213,11 @@ skipIfNoFFmpeg = _skipIf( ...@@ -212,3 +213,11 @@ skipIfNoFFmpeg = _skipIf(
reason="ffmpeg features are not available.", reason="ffmpeg features are not available.",
key="NO_FFMPEG", key="NO_FFMPEG",
) )
skipIfPy310 = _skipIf(
sys.version_info >= (3, 10, 0),
reason=(
"Test is known to fail for Python 3.10, disabling for now"
"See: https://github.com/pytorch/audio/pull/2224#issuecomment-1048329450"
),
key="ON_PYTHON_310",
)
test/torchaudio_unittest/models/tacotron2/model_test_impl.py
View file @ 27dff6ba
...@@ -5,7 +5,7 @@ from parameterized import parameterized ...@@ -5,7 +5,7 @@ from parameterized import parameterized
from torch import Tensor from torch import Tensor
from torchaudio.models import Tacotron2 from torchaudio.models import Tacotron2
from torchaudio.models.tacotron2 import _Encoder, _Decoder from torchaudio.models.tacotron2 import _Encoder, _Decoder
from torchaudio_unittest.common_utils import TestBaseMixin, torch_script from torchaudio_unittest.common_utils import TestBaseMixin, torch_script, skipIfPy310
class Tacotron2InferenceWrapper(torch.nn.Module): class Tacotron2InferenceWrapper(torch.nn.Module):
...@@ -42,6 +42,7 @@ class TorchscriptConsistencyMixin(TestBaseMixin): ...@@ -42,6 +42,7 @@ class TorchscriptConsistencyMixin(TestBaseMixin):
class Tacotron2EncoderTests(TorchscriptConsistencyMixin): class Tacotron2EncoderTests(TorchscriptConsistencyMixin):
@skipIfPy310
def test_tacotron2_torchscript_consistency(self): def test_tacotron2_torchscript_consistency(self):
r"""Validate the torchscript consistency of a Encoder.""" r"""Validate the torchscript consistency of a Encoder."""
n_batch, n_seq, encoder_embedding_dim = 16, 64, 512 n_batch, n_seq, encoder_embedding_dim = 16, 64, 512
...@@ -265,6 +266,7 @@ class Tacotron2Tests(TorchscriptConsistencyMixin): ...@@ -265,6 +266,7 @@ class Tacotron2Tests(TorchscriptConsistencyMixin):
(16,), (16,),
] ]
) )
@skipIfPy310
def test_tacotron2_torchscript_consistency(self, n_batch): def test_tacotron2_torchscript_consistency(self, n_batch):
r"""Validate the torchscript consistency of a Tacotron2.""" r"""Validate the torchscript consistency of a Tacotron2."""
n_mels = 80 n_mels = 80
...@@ -333,6 +335,7 @@ class Tacotron2Tests(TorchscriptConsistencyMixin): ...@@ -333,6 +335,7 @@ class Tacotron2Tests(TorchscriptConsistencyMixin):
(16,), (16,),
] ]
) )
@skipIfPy310
def test_tacotron2_inference_torchscript_consistency(self, n_batch): def test_tacotron2_inference_torchscript_consistency(self, n_batch):
r"""Validate the torchscript consistency of Tacotron2 inference function.""" r"""Validate the torchscript consistency of Tacotron2 inference function."""
n_mels = 40 n_mels = 40
......
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