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:
command: .circleci/unittest/linux/scripts/run_test.sh
environment:
TORCHAUDIO_TEST_ALLOW_SKIP_IF_NO_CUDA: true
TORCHAUDIO_TEST_ALLOW_SKIP_IF_ON_PYTHON_310: true
- store_test_results:
path: test-results
- store_artifacts:
......@@ -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
- run:
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:
path: test-results
- store_artifacts:
......@@ -540,6 +544,8 @@ jobs:
TORCHAUDIO_TEST_ALLOW_SKIP_IF_NO_CUDA: true
TORCHAUDIO_TEST_ALLOW_SKIP_IF_NO_KALDI: 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:
path: test-results
- store_artifacts:
......@@ -583,6 +589,8 @@ jobs:
TORCHAUDIO_TEST_ALLOW_SKIP_IF_NO_CTC_DECODER: true
TORCHAUDIO_TEST_ALLOW_SKIP_IF_NO_KALDI: 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:
path: test-results
- store_artifacts:
......@@ -618,6 +626,8 @@ jobs:
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_QUANTIZATION: true
TORCHAUDIO_TEST_ALLOW_SKIP_IF_ON_PYTHON_310: true
TORCHAUDIO_TEST_ALLOW_SKIP_IF_NO_MOD_sentencepiece: true
- store_test_results:
path: test-results
- store_artifacts:
......@@ -869,6 +879,54 @@ workflows:
requires:
- download_third_parties_nix
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:
cuda_version: cpu
name: binary_macos_wheel_py3.7_cpu
......@@ -887,6 +945,12 @@ workflows:
python_version: '3.9'
requires:
- 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:
cuda_version: cpu
name: binary_windows_wheel_py3.7_cpu
......@@ -929,6 +993,20 @@ workflows:
name: binary_windows_wheel_py3.9_cu115
python_version: '3.9'
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:
conda_docker_image: pytorch/conda-builder:cpu
cuda_version: cpu
......@@ -1034,6 +1112,41 @@ workflows:
python_version: '3.9'
requires:
- 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:
conda_docker_image: pytorch/conda-builder:cpu
cuda_version: cpu
......@@ -1055,6 +1168,13 @@ workflows:
python_version: '3.9'
requires:
- 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:
conda_docker_image: pytorch/conda-builder:cpu
cuda_version: cpu
......@@ -1100,6 +1220,21 @@ workflows:
cuda_version: cu115
name: binary_windows_conda_py3.9_cu115
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:
filters:
branches:
......@@ -1154,6 +1289,12 @@ workflows:
python_version: '3.9'
requires:
- 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:
cuda_version: cu113
name: unittest_linux_gpu_py3.7
......@@ -1172,6 +1313,12 @@ workflows:
python_version: '3.9'
requires:
- 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:
cuda_version: cpu
name: unittest_windows_cpu_py3.7
......@@ -1184,6 +1331,10 @@ workflows:
cuda_version: cpu
name: unittest_windows_cpu_py3.9
python_version: '3.9'
- unittest_windows_cpu:
cuda_version: cpu
name: unittest_windows_cpu_py3.10
python_version: '3.10'
- unittest_windows_gpu:
cuda_version: cu113
name: unittest_windows_gpu_py3.7
......@@ -1196,6 +1347,10 @@ workflows:
cuda_version: cu113
name: unittest_windows_gpu_py3.9
python_version: '3.9'
- unittest_windows_gpu:
cuda_version: cu113
name: unittest_windows_gpu_py3.10
python_version: '3.10'
- unittest_macos_cpu:
cuda_version: cpu
name: unittest_macos_cpu_py3.7
......@@ -1214,6 +1369,12 @@ workflows:
python_version: '3.9'
requires:
- 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:
jobs:
- circleci_consistency:
......@@ -2002,7 +2163,7 @@ workflows:
python_version: '3.9'
requires:
- nightly_binary_linux_wheel_py3.9_rocm4.5.2_upload
- binary_macos_wheel:
- binary_linux_wheel:
cuda_version: cpu
filters:
branches:
......@@ -2010,8 +2171,8 @@ workflows:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_macos_wheel_py3.7_cpu
python_version: '3.7'
name: nightly_binary_linux_wheel_py3.10_cpu
python_version: '3.10'
requires:
- download_third_parties_nix
- binary_wheel_upload:
......@@ -2022,11 +2183,11 @@ workflows:
- nightly
tags:
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:
- nightly_binary_macos_wheel_py3.7_cpu
subfolder: ''
- binary_macos_wheel:
- nightly_binary_linux_wheel_py3.10_cpu
subfolder: cpu/
- smoke_test_linux_pip:
cuda_version: cpu
filters:
branches:
......@@ -2034,56 +2195,60 @@ workflows:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_macos_wheel_py3.8_cpu
python_version: '3.8'
name: nightly_binary_linux_wheel_py3.10_cpu_smoke_test_pip
python_version: '3.10'
requires:
- download_third_parties_nix
- binary_wheel_upload:
context: org-member
- nightly_binary_linux_wheel_py3.10_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_macos_wheel_py3.8_cpu_upload
name: nightly_binary_linux_wheel_py3.10_cu102
python_version: '3.10'
requires:
- nightly_binary_macos_wheel_py3.8_cpu
subfolder: ''
- binary_macos_wheel:
cuda_version: cpu
- 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_macos_wheel_py3.9_cpu
python_version: '3.9'
name: nightly_binary_linux_wheel_py3.10_cu102_upload
requires:
- download_third_parties_nix
- binary_wheel_upload:
context: org-member
- nightly_binary_linux_wheel_py3.10_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_macos_wheel_py3.9_cpu_upload
name: nightly_binary_linux_wheel_py3.10_cu102_smoke_test_pip
python_version: '3.10'
requires:
- nightly_binary_macos_wheel_py3.9_cpu
subfolder: ''
- binary_windows_wheel:
cuda_version: cpu
- nightly_binary_linux_wheel_py3.10_cu102_upload
- binary_linux_wheel:
cuda_version: cu111
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'
name: nightly_binary_linux_wheel_py3.10_cu111
python_version: '3.10'
requires:
- download_third_parties_nix
wheel_docker_image: pytorch/manylinux-cuda111
- binary_wheel_upload:
context: org-member
filters:
......@@ -2092,23 +2257,23 @@ workflows:
- nightly
tags:
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:
- nightly_binary_windows_wheel_py3.7_cpu
subfolder: cpu/
- smoke_test_windows_pip:
cuda_version: cpu
- nightly_binary_linux_wheel_py3.10_cu111
subfolder: cu111/
- smoke_test_linux_pip:
cuda_version: cu111
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'
name: nightly_binary_linux_wheel_py3.10_cu111_smoke_test_pip
python_version: '3.10'
requires:
- nightly_binary_windows_wheel_py3.7_cpu_upload
- binary_windows_wheel:
- nightly_binary_linux_wheel_py3.10_cu111_upload
- binary_linux_wheel:
cuda_version: cu113
filters:
branches:
......@@ -2116,8 +2281,10 @@ workflows:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_windows_wheel_py3.7_cu113
python_version: '3.7'
name: nightly_binary_linux_wheel_py3.10_cu113
python_version: '3.10'
requires:
- download_third_parties_nix
wheel_docker_image: pytorch/manylinux-cuda113
- binary_wheel_upload:
context: org-member
......@@ -2127,11 +2294,11 @@ workflows:
- nightly
tags:
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:
- nightly_binary_windows_wheel_py3.7_cu113
- nightly_binary_linux_wheel_py3.10_cu113
subfolder: cu113/
- smoke_test_windows_pip:
- smoke_test_linux_pip:
cuda_version: cu113
filters:
branches:
......@@ -2139,11 +2306,11 @@ workflows:
- 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'
name: nightly_binary_linux_wheel_py3.10_cu113_smoke_test_pip
python_version: '3.10'
requires:
- nightly_binary_windows_wheel_py3.7_cu113_upload
- binary_windows_wheel:
- nightly_binary_linux_wheel_py3.10_cu113_upload
- binary_linux_wheel:
cuda_version: cu115
filters:
branches:
......@@ -2151,8 +2318,10 @@ workflows:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_windows_wheel_py3.7_cu115
python_version: '3.7'
name: nightly_binary_linux_wheel_py3.10_cu115
python_version: '3.10'
requires:
- download_third_parties_nix
wheel_docker_image: pytorch/manylinux-cuda115
- binary_wheel_upload:
context: org-member
......@@ -2162,11 +2331,11 @@ workflows:
- nightly
tags:
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:
- nightly_binary_windows_wheel_py3.7_cu115
- nightly_binary_linux_wheel_py3.10_cu115
subfolder: cu115/
- smoke_test_windows_pip:
- smoke_test_linux_pip:
cuda_version: cu115
filters:
branches:
......@@ -2174,20 +2343,23 @@ workflows:
- 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'
name: nightly_binary_linux_wheel_py3.10_cu115_smoke_test_pip
python_version: '3.10'
requires:
- nightly_binary_windows_wheel_py3.7_cu115_upload
- binary_windows_wheel:
cuda_version: cpu
- nightly_binary_linux_wheel_py3.10_cu115_upload
- binary_linux_wheel:
cuda_version: rocm4.3.1
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'
name: nightly_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_wheel_upload:
context: org-member
filters:
......@@ -2196,33 +2368,35 @@ workflows:
- nightly
tags:
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:
- nightly_binary_windows_wheel_py3.8_cpu
subfolder: cpu/
- smoke_test_windows_pip:
cuda_version: cpu
- nightly_binary_linux_wheel_py3.10_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_windows_wheel_py3.8_cpu_smoke_test_pip
python_version: '3.8'
name: nightly_binary_linux_wheel_py3.10_rocm4.3.1_smoke_test_pip
python_version: '3.10'
requires:
- nightly_binary_windows_wheel_py3.8_cpu_upload
- binary_windows_wheel:
cuda_version: cu113
- nightly_binary_linux_wheel_py3.10_rocm4.3.1_upload
- binary_linux_wheel:
cuda_version: rocm4.5.2
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
name: nightly_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_wheel_upload:
context: org-member
filters:
......@@ -2231,33 +2405,34 @@ workflows:
- nightly
tags:
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:
- nightly_binary_windows_wheel_py3.8_cu113
subfolder: cu113/
- smoke_test_windows_pip:
cuda_version: cu113
- nightly_binary_linux_wheel_py3.10_rocm4.5.2
subfolder: rocm4.5.2/
- smoke_test_linux_pip:
cuda_version: rocm4.5.2
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'
name: nightly_binary_linux_wheel_py3.10_rocm4.5.2_smoke_test_pip
python_version: '3.10'
requires:
- nightly_binary_windows_wheel_py3.8_cu113_upload
- binary_windows_wheel:
cuda_version: cu115
- nightly_binary_linux_wheel_py3.10_rocm4.5.2_upload
- binary_macos_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_cu115
python_version: '3.8'
wheel_docker_image: pytorch/manylinux-cuda115
name: nightly_binary_macos_wheel_py3.7_cpu
python_version: '3.7'
requires:
- download_third_parties_nix
- binary_wheel_upload:
context: org-member
filters:
......@@ -2266,24 +2441,292 @@ workflows:
- nightly
tags:
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:
- nightly_binary_windows_wheel_py3.8_cu115
subfolder: cu115/
- smoke_test_windows_pip:
cuda_version: cu115
- nightly_binary_macos_wheel_py3.7_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_windows_wheel_py3.8_cu115_smoke_test_pip
name: nightly_binary_macos_wheel_py3.8_cpu
python_version: '3.8'
requires:
- nightly_binary_windows_wheel_py3.8_cu115_upload
- binary_windows_wheel:
cuda_version: cpu
- 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.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:
branches:
only:
......@@ -2347,10 +2790,114 @@ workflows:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_windows_wheel_py3.9_cu113_smoke_test_pip
python_version: '3.9'
name: nightly_binary_windows_wheel_py3.9_cu113_smoke_test_pip
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:
- nightly_binary_windows_wheel_py3.9_cu113_upload
- nightly_binary_windows_wheel_py3.10_cu113_upload
- binary_windows_wheel:
cuda_version: cu115
filters:
......@@ -2359,8 +2906,8 @@ workflows:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_windows_wheel_py3.9_cu115
python_version: '3.9'
name: nightly_binary_windows_wheel_py3.10_cu115
python_version: '3.10'
wheel_docker_image: pytorch/manylinux-cuda115
- binary_wheel_upload:
context: org-member
......@@ -2370,9 +2917,9 @@ workflows:
- nightly
tags:
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:
- nightly_binary_windows_wheel_py3.9_cu115
- nightly_binary_windows_wheel_py3.10_cu115
subfolder: cu115/
- smoke_test_windows_pip:
cuda_version: cu115
......@@ -2382,10 +2929,10 @@ workflows:
- 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'
name: nightly_binary_windows_wheel_py3.10_cu115_smoke_test_pip
python_version: '3.10'
requires:
- nightly_binary_windows_wheel_py3.9_cu115_upload
- nightly_binary_windows_wheel_py3.10_cu115_upload
- binary_linux_conda:
conda_docker_image: pytorch/conda-builder:cpu
cuda_version: cpu
......@@ -2926,6 +3473,186 @@ workflows:
python_version: '3.9'
requires:
- 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:
conda_docker_image: pytorch/conda-builder:cpu
cuda_version: cpu
......@@ -2998,6 +3725,30 @@ workflows:
name: nightly_binary_macos_conda_py3.9_cpu_upload
requires:
- 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:
conda_docker_image: pytorch/conda-builder:cpu
cuda_version: cpu
......@@ -3304,3 +4055,105 @@ workflows:
python_version: '3.9'
requires:
- 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:
command: .circleci/unittest/linux/scripts/run_test.sh
environment:
TORCHAUDIO_TEST_ALLOW_SKIP_IF_NO_CUDA: true
TORCHAUDIO_TEST_ALLOW_SKIP_IF_ON_PYTHON_310: true
- store_test_results:
path: test-results
- store_artifacts:
......@@ -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
- run:
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:
path: test-results
- store_artifacts:
......@@ -540,6 +544,8 @@ jobs:
TORCHAUDIO_TEST_ALLOW_SKIP_IF_NO_CUDA: true
TORCHAUDIO_TEST_ALLOW_SKIP_IF_NO_KALDI: 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:
path: test-results
- store_artifacts:
......@@ -583,6 +589,8 @@ jobs:
TORCHAUDIO_TEST_ALLOW_SKIP_IF_NO_CTC_DECODER: true
TORCHAUDIO_TEST_ALLOW_SKIP_IF_NO_KALDI: 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:
path: test-results
- store_artifacts:
......@@ -618,6 +626,8 @@ jobs:
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_QUANTIZATION: true
TORCHAUDIO_TEST_ALLOW_SKIP_IF_ON_PYTHON_310: true
TORCHAUDIO_TEST_ALLOW_SKIP_IF_NO_MOD_sentencepiece: true
- store_test_results:
path: test-results
- store_artifacts:
......
.circleci/regenerate.py
View file @ 27dff6ba
......@@ -21,7 +21,7 @@ import yaml
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 = {
"linux": ["cpu", "cu102", "cu111", "cu113", "cu115", "rocm4.3.1", "rocm4.5.2"],
"windows": ["cpu", "cu113", "cu115"],
......
.circleci/unittest/linux/scripts/install.sh
View file @ 27dff6ba
......@@ -37,6 +37,10 @@ fi
printf "Installing PyTorch with %s\n" "${cudatoolkit}"
(
if [[ "$(python --version)" = *3.10* ]]; then
CONDA_CHANNEL_FLAGS="-c conda-forge"
fi
if [ "${os}" == MacOSX ] ; then
# TODO: this can be removed as soon as linking issue could be resolved
# see https://github.com/pytorch/pytorch/issues/62424 from details
......@@ -57,8 +61,8 @@ python setup.py install
# 3. Install Test tools
printf "* Installing test tools\n"
NUMBA_DEV_CHANNEL=""
if [[ "$(python --version)" = *3.9* ]]; then
# Numba isn't available for Python 3.9 except on the numba dev channel and building from source fails
if [[ "$(python --version)" = *3.9* || "$(python --version)" = *3.10* ]]; then
# 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
NUMBA_DEV_CHANNEL="-c numba/label/dev"
fi
......
.circleci/unittest/linux/scripts/setup_env.sh
View file @ 27dff6ba
......@@ -25,7 +25,15 @@ if [ ! -d "${conda_dir}" ]; then
bash ./miniconda.sh -b -f -p "${conda_dir}"
eval "$("${conda_dir}/bin/conda" shell.bash hook)"
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
eval "$("${conda_dir}/bin/conda" shell.bash hook)"
fi
......
.circleci/unittest/windows/scripts/install.sh
View file @ 27dff6ba
......@@ -48,16 +48,39 @@ printf "* Installing torchaudio\n"
# 3. Install Test tools
printf "* Installing test tools\n"
NUMBA_DEV_CHANNEL=""
if [[ "$(python --version)" = *3.9* ]]; then
# Numba isn't available for Python 3.9 except on the numba dev channel and building from source fails
# See https://github.com/librosa/librosa/issues/1270#issuecomment-759065048
NUMBA_DEV_CHANNEL="-c numba/label/dev"
fi
SENTENCEPIECE_DEPENDENCY="sentencepiece"
case "$(python --version)" in
*3.9*)
# Numba isn't available for Python 3.9 except on the numba dev channel and building from source fails
# 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.
(
set -x
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
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
unset miniconda_exe
eval "$("${conda_dir}/Scripts/conda.exe" 'shell.bash' 'hook')"
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
eval "$("${conda_dir}/Scripts/conda.exe" 'shell.bash' 'hook')"
fi
......
packaging/build_conda.sh
View file @ 27dff6ba
......@@ -12,9 +12,15 @@ setup_conda_cudatoolkit_constraint
setup_visual_studio_constraint
# 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"
if [[ "$CU_VERSION" == cu115 ]]; then
export CUDATOOLKIT_CHANNEL="conda-forge"
# NOTE: This is needed because `cudatoolkit=11.5` has a dependency on 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
conda build -c defaults -c $CUDATOOLKIT_CHANNEL $CONDA_CHANNEL_FLAGS --no-anaconda-upload --python "$PYTHON_VERSION" packaging/torchaudio
# NOTE: There are some dependencies that are not available for macOS on Python 3.10 without conda-forge
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() {
3.7) python_abi=cp37-cp37m ;;
3.8) python_abi=cp38-cp38 ;;
3.9) python_abi=cp39-cp39 ;;
3.10) python_abi=cp310-cp310 ;;
*)
echo "Unrecognized PYTHON_VERSION=$PYTHON_VERSION"
exit 1
......
packaging/torchaudio/meta.yaml
View file @ 27dff6ba
......@@ -17,7 +17,8 @@ requirements:
- pkg-config # [not win]
- cmake
- ninja
- defaults::numpy >=1.11
- numpy>=1.11 # [py <= 39]
- numpy>=1.21.2 # [py >= 310]
- pytorch-mutex 1.0 {{ build_variant }} # [not osx ]
{{ environ.get('CONDA_PYTORCH_BUILD_CONSTRAINT', 'pytorch') }}
{{ environ.get('CONDA_EXTRA_BUILD_CONSTRAINT', '') }}
......@@ -26,7 +27,8 @@ requirements:
run:
- python
- defaults::numpy >=1.11
- numpy>=1.11 # [py <= 39]
- numpy>=1.21.2 # [py >= 310]
- pytorch-mutex 1.0 {{ build_variant }} # [not osx ]
{{ environ.get('CONDA_PYTORCH_CONSTRAINT', 'pytorch') }}
{{ environ.get('CONDA_CUDATOOLKIT_CONSTRAINT', '') }}
......
test/torchaudio_unittest/common_utils/__init__.py
View file @ 27dff6ba
......@@ -17,6 +17,7 @@ from .case_utils import (
skipIfRocm,
skipIfNoQengine,
skipIfNoFFmpeg,
skipIfPy310,
)
from .data_utils import (
get_asset_path,
......@@ -59,6 +60,7 @@ __all__ = [
"skipIfRocm",
"skipIfNoQengine",
"skipIfNoFFmpeg",
"skipIfPy310",
"get_wav_data",
"normalize_wav",
"load_wav",
......
test/torchaudio_unittest/common_utils/case_utils.py
View file @ 27dff6ba
......@@ -2,6 +2,7 @@ import functools
import os.path
import shutil
import subprocess
import sys
import tempfile
import time
import unittest
......@@ -212,3 +213,11 @@ skipIfNoFFmpeg = _skipIf(
reason="ffmpeg features are not available.",
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
from torch import Tensor
from torchaudio.models import Tacotron2
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):
......@@ -42,6 +42,7 @@ class TorchscriptConsistencyMixin(TestBaseMixin):
class Tacotron2EncoderTests(TorchscriptConsistencyMixin):
@skipIfPy310
def test_tacotron2_torchscript_consistency(self):
r"""Validate the torchscript consistency of a Encoder."""
n_batch, n_seq, encoder_embedding_dim = 16, 64, 512
......@@ -265,6 +266,7 @@ class Tacotron2Tests(TorchscriptConsistencyMixin):
(16,),
]
)
@skipIfPy310
def test_tacotron2_torchscript_consistency(self, n_batch):
r"""Validate the torchscript consistency of a Tacotron2."""
n_mels = 80
......@@ -333,6 +335,7 @@ class Tacotron2Tests(TorchscriptConsistencyMixin):
(16,),
]
)
@skipIfPy310
def test_tacotron2_inference_torchscript_consistency(self, n_batch):
r"""Validate the torchscript consistency of Tacotron2 inference function."""
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