Unverified Commit a85f1adc authored by Nikita Shulga's avatar Nikita Shulga Committed by GitHub
Browse files

Limit Windows GPU testing to CUDA-11.3 only (#1842)

* Limit Windows GPU testing to CUDA-11.3 only

Which is the only CUDA version that planned to be supported on Windows
for the upcoming release

* Move unittests to 11.3 as well
parent 19f7f971
...@@ -826,57 +826,37 @@ workflows: ...@@ -826,57 +826,37 @@ workflows:
name: binary_windows_wheel_py3.6_cpu name: binary_windows_wheel_py3.6_cpu
python_version: '3.6' python_version: '3.6'
- binary_windows_wheel: - binary_windows_wheel:
cuda_version: cu102 cuda_version: cu113
name: binary_windows_wheel_py3.6_cu102 name: binary_windows_wheel_py3.6_cu113
python_version: '3.6'
wheel_docker_image: pytorch/manylinux-cuda102
- binary_windows_wheel:
cuda_version: cu111
name: binary_windows_wheel_py3.6_cu111
python_version: '3.6' python_version: '3.6'
wheel_docker_image: pytorch/manylinux-cuda111 wheel_docker_image: pytorch/manylinux-cuda113
- 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
python_version: '3.7' python_version: '3.7'
- binary_windows_wheel: - binary_windows_wheel:
cuda_version: cu102 cuda_version: cu113
name: binary_windows_wheel_py3.7_cu102 name: binary_windows_wheel_py3.7_cu113
python_version: '3.7'
wheel_docker_image: pytorch/manylinux-cuda102
- binary_windows_wheel:
cuda_version: cu111
name: binary_windows_wheel_py3.7_cu111
python_version: '3.7' python_version: '3.7'
wheel_docker_image: pytorch/manylinux-cuda111 wheel_docker_image: pytorch/manylinux-cuda113
- binary_windows_wheel: - binary_windows_wheel:
cuda_version: cpu cuda_version: cpu
name: binary_windows_wheel_py3.8_cpu name: binary_windows_wheel_py3.8_cpu
python_version: '3.8' python_version: '3.8'
- binary_windows_wheel: - binary_windows_wheel:
cuda_version: cu102 cuda_version: cu113
name: binary_windows_wheel_py3.8_cu102 name: binary_windows_wheel_py3.8_cu113
python_version: '3.8'
wheel_docker_image: pytorch/manylinux-cuda102
- binary_windows_wheel:
cuda_version: cu111
name: binary_windows_wheel_py3.8_cu111
python_version: '3.8' python_version: '3.8'
wheel_docker_image: pytorch/manylinux-cuda111 wheel_docker_image: pytorch/manylinux-cuda113
- binary_windows_wheel: - binary_windows_wheel:
cuda_version: cpu cuda_version: cpu
name: binary_windows_wheel_py3.9_cpu name: binary_windows_wheel_py3.9_cpu
python_version: '3.9' python_version: '3.9'
- binary_windows_wheel: - binary_windows_wheel:
cuda_version: cu102 cuda_version: cu113
name: binary_windows_wheel_py3.9_cu102 name: binary_windows_wheel_py3.9_cu113
python_version: '3.9'
wheel_docker_image: pytorch/manylinux-cuda102
- binary_windows_wheel:
cuda_version: cu111
name: binary_windows_wheel_py3.9_cu111
python_version: '3.9' python_version: '3.9'
wheel_docker_image: pytorch/manylinux-cuda111 wheel_docker_image: pytorch/manylinux-cuda113
- 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
...@@ -1023,14 +1003,9 @@ workflows: ...@@ -1023,14 +1003,9 @@ workflows:
name: binary_windows_conda_py3.6_cpu name: binary_windows_conda_py3.6_cpu
python_version: '3.6' python_version: '3.6'
- binary_windows_conda: - binary_windows_conda:
conda_docker_image: pytorch/conda-builder:cuda102 conda_docker_image: pytorch/conda-builder:cuda113
cuda_version: cu102 cuda_version: cu113
name: binary_windows_conda_py3.6_cu102 name: binary_windows_conda_py3.6_cu113
python_version: '3.6'
- binary_windows_conda:
conda_docker_image: pytorch/conda-builder:cuda111
cuda_version: cu111
name: binary_windows_conda_py3.6_cu111
python_version: '3.6' python_version: '3.6'
- binary_windows_conda: - binary_windows_conda:
conda_docker_image: pytorch/conda-builder:cpu conda_docker_image: pytorch/conda-builder:cpu
...@@ -1038,14 +1013,9 @@ workflows: ...@@ -1038,14 +1013,9 @@ workflows:
name: binary_windows_conda_py3.7_cpu name: binary_windows_conda_py3.7_cpu
python_version: '3.7' python_version: '3.7'
- binary_windows_conda: - binary_windows_conda:
conda_docker_image: pytorch/conda-builder:cuda102 conda_docker_image: pytorch/conda-builder:cuda113
cuda_version: cu102 cuda_version: cu113
name: binary_windows_conda_py3.7_cu102 name: binary_windows_conda_py3.7_cu113
python_version: '3.7'
- binary_windows_conda:
conda_docker_image: pytorch/conda-builder:cuda111
cuda_version: cu111
name: binary_windows_conda_py3.7_cu111
python_version: '3.7' python_version: '3.7'
- binary_windows_conda: - binary_windows_conda:
conda_docker_image: pytorch/conda-builder:cpu conda_docker_image: pytorch/conda-builder:cpu
...@@ -1053,14 +1023,9 @@ workflows: ...@@ -1053,14 +1023,9 @@ workflows:
name: binary_windows_conda_py3.8_cpu name: binary_windows_conda_py3.8_cpu
python_version: '3.8' python_version: '3.8'
- binary_windows_conda: - binary_windows_conda:
conda_docker_image: pytorch/conda-builder:cuda102 conda_docker_image: pytorch/conda-builder:cuda113
cuda_version: cu102 cuda_version: cu113
name: binary_windows_conda_py3.8_cu102 name: binary_windows_conda_py3.8_cu113
python_version: '3.8'
- binary_windows_conda:
conda_docker_image: pytorch/conda-builder:cuda111
cuda_version: cu111
name: binary_windows_conda_py3.8_cu111
python_version: '3.8' python_version: '3.8'
- binary_windows_conda: - binary_windows_conda:
conda_docker_image: pytorch/conda-builder:cpu conda_docker_image: pytorch/conda-builder:cpu
...@@ -1068,14 +1033,9 @@ workflows: ...@@ -1068,14 +1033,9 @@ workflows:
name: binary_windows_conda_py3.9_cpu name: binary_windows_conda_py3.9_cpu
python_version: '3.9' python_version: '3.9'
- binary_windows_conda: - binary_windows_conda:
conda_docker_image: pytorch/conda-builder:cuda102 conda_docker_image: pytorch/conda-builder:cuda113
cuda_version: cu102 cuda_version: cu113
name: binary_windows_conda_py3.9_cu102 name: binary_windows_conda_py3.9_cu113
python_version: '3.9'
- binary_windows_conda:
conda_docker_image: pytorch/conda-builder:cuda111
cuda_version: cu111
name: binary_windows_conda_py3.9_cu111
python_version: '3.9' python_version: '3.9'
- build_docs: - build_docs:
filters: filters:
...@@ -1138,25 +1098,25 @@ workflows: ...@@ -1138,25 +1098,25 @@ workflows:
requires: requires:
- download_third_parties_nix - download_third_parties_nix
- unittest_linux_gpu: - unittest_linux_gpu:
cuda_version: cu102 cuda_version: cu113
name: unittest_linux_gpu_py3.6 name: unittest_linux_gpu_py3.6
python_version: '3.6' python_version: '3.6'
requires: requires:
- download_third_parties_nix - download_third_parties_nix
- unittest_linux_gpu: - unittest_linux_gpu:
cuda_version: cu102 cuda_version: cu113
name: unittest_linux_gpu_py3.7 name: unittest_linux_gpu_py3.7
python_version: '3.7' python_version: '3.7'
requires: requires:
- download_third_parties_nix - download_third_parties_nix
- unittest_linux_gpu: - unittest_linux_gpu:
cuda_version: cu102 cuda_version: cu113
name: unittest_linux_gpu_py3.8 name: unittest_linux_gpu_py3.8
python_version: '3.8' python_version: '3.8'
requires: requires:
- download_third_parties_nix - download_third_parties_nix
- unittest_linux_gpu: - unittest_linux_gpu:
cuda_version: cu102 cuda_version: cu113
name: unittest_linux_gpu_py3.9 name: unittest_linux_gpu_py3.9
python_version: '3.9' python_version: '3.9'
requires: requires:
...@@ -1178,19 +1138,19 @@ workflows: ...@@ -1178,19 +1138,19 @@ workflows:
name: unittest_windows_cpu_py3.9 name: unittest_windows_cpu_py3.9
python_version: '3.9' python_version: '3.9'
- unittest_windows_gpu: - unittest_windows_gpu:
cuda_version: cu102 cuda_version: cu113
name: unittest_windows_gpu_py3.6 name: unittest_windows_gpu_py3.6
python_version: '3.6' python_version: '3.6'
- unittest_windows_gpu: - unittest_windows_gpu:
cuda_version: cu102 cuda_version: cu113
name: unittest_windows_gpu_py3.7 name: unittest_windows_gpu_py3.7
python_version: '3.7' python_version: '3.7'
- unittest_windows_gpu: - unittest_windows_gpu:
cuda_version: cu102 cuda_version: cu113
name: unittest_windows_gpu_py3.8 name: unittest_windows_gpu_py3.8
python_version: '3.8' python_version: '3.8'
- unittest_windows_gpu: - unittest_windows_gpu:
cuda_version: cu102 cuda_version: cu113
name: unittest_windows_gpu_py3.9 name: unittest_windows_gpu_py3.9
python_version: '3.9' python_version: '3.9'
- unittest_macos_cpu: - unittest_macos_cpu:
...@@ -2098,51 +2058,16 @@ workflows: ...@@ -2098,51 +2058,16 @@ workflows:
requires: requires:
- nightly_binary_windows_wheel_py3.6_cpu_upload - nightly_binary_windows_wheel_py3.6_cpu_upload
- binary_windows_wheel: - binary_windows_wheel:
cuda_version: cu102 cuda_version: cu113
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_windows_wheel_py3.6_cu102
python_version: '3.6'
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_windows_wheel_py3.6_cu102_upload
requires:
- nightly_binary_windows_wheel_py3.6_cu102
subfolder: cu102/
- smoke_test_windows_pip:
cuda_version: cu102
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_windows_wheel_py3.6_cu102_smoke_test_pip
python_version: '3.6'
requires:
- nightly_binary_windows_wheel_py3.6_cu102_upload
- binary_windows_wheel:
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.6_cu111 name: nightly_binary_windows_wheel_py3.6_cu113
python_version: '3.6' python_version: '3.6'
wheel_docker_image: pytorch/manylinux-cuda111 wheel_docker_image: pytorch/manylinux-cuda113
- binary_wheel_upload: - binary_wheel_upload:
context: org-member context: org-member
filters: filters:
...@@ -2151,22 +2076,22 @@ workflows: ...@@ -2151,22 +2076,22 @@ workflows:
- nightly - nightly
tags: tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_windows_wheel_py3.6_cu111_upload name: nightly_binary_windows_wheel_py3.6_cu113_upload
requires: requires:
- nightly_binary_windows_wheel_py3.6_cu111 - nightly_binary_windows_wheel_py3.6_cu113
subfolder: cu111/ subfolder: cu113/
- smoke_test_windows_pip: - smoke_test_windows_pip:
cuda_version: cu111 cuda_version: cu113
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.6_cu111_smoke_test_pip name: nightly_binary_windows_wheel_py3.6_cu113_smoke_test_pip
python_version: '3.6' python_version: '3.6'
requires: requires:
- nightly_binary_windows_wheel_py3.6_cu111_upload - nightly_binary_windows_wheel_py3.6_cu113_upload
- binary_windows_wheel: - binary_windows_wheel:
cuda_version: cpu cuda_version: cpu
filters: filters:
...@@ -2202,51 +2127,16 @@ workflows: ...@@ -2202,51 +2127,16 @@ workflows:
requires: requires:
- nightly_binary_windows_wheel_py3.7_cpu_upload - nightly_binary_windows_wheel_py3.7_cpu_upload
- binary_windows_wheel: - binary_windows_wheel:
cuda_version: cu102 cuda_version: cu113
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_windows_wheel_py3.7_cu102
python_version: '3.7'
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_windows_wheel_py3.7_cu102_upload
requires:
- nightly_binary_windows_wheel_py3.7_cu102
subfolder: cu102/
- smoke_test_windows_pip:
cuda_version: cu102
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_windows_wheel_py3.7_cu102_smoke_test_pip
python_version: '3.7'
requires:
- nightly_binary_windows_wheel_py3.7_cu102_upload
- binary_windows_wheel:
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_cu111 name: nightly_binary_windows_wheel_py3.7_cu113
python_version: '3.7' python_version: '3.7'
wheel_docker_image: pytorch/manylinux-cuda111 wheel_docker_image: pytorch/manylinux-cuda113
- binary_wheel_upload: - binary_wheel_upload:
context: org-member context: org-member
filters: filters:
...@@ -2255,22 +2145,22 @@ workflows: ...@@ -2255,22 +2145,22 @@ workflows:
- nightly - nightly
tags: tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_windows_wheel_py3.7_cu111_upload name: nightly_binary_windows_wheel_py3.7_cu113_upload
requires: requires:
- nightly_binary_windows_wheel_py3.7_cu111 - nightly_binary_windows_wheel_py3.7_cu113
subfolder: cu111/ subfolder: cu113/
- smoke_test_windows_pip: - smoke_test_windows_pip:
cuda_version: cu111 cuda_version: cu113
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_cu111_smoke_test_pip name: nightly_binary_windows_wheel_py3.7_cu113_smoke_test_pip
python_version: '3.7' python_version: '3.7'
requires: requires:
- nightly_binary_windows_wheel_py3.7_cu111_upload - nightly_binary_windows_wheel_py3.7_cu113_upload
- binary_windows_wheel: - binary_windows_wheel:
cuda_version: cpu cuda_version: cpu
filters: filters:
...@@ -2306,51 +2196,16 @@ workflows: ...@@ -2306,51 +2196,16 @@ workflows:
requires: requires:
- nightly_binary_windows_wheel_py3.8_cpu_upload - nightly_binary_windows_wheel_py3.8_cpu_upload
- binary_windows_wheel: - binary_windows_wheel:
cuda_version: cu102 cuda_version: cu113
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_windows_wheel_py3.8_cu102
python_version: '3.8'
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_windows_wheel_py3.8_cu102_upload
requires:
- nightly_binary_windows_wheel_py3.8_cu102
subfolder: cu102/
- smoke_test_windows_pip:
cuda_version: cu102
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_windows_wheel_py3.8_cu102_smoke_test_pip
python_version: '3.8'
requires:
- nightly_binary_windows_wheel_py3.8_cu102_upload
- binary_windows_wheel:
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.8_cu111 name: nightly_binary_windows_wheel_py3.8_cu113
python_version: '3.8' python_version: '3.8'
wheel_docker_image: pytorch/manylinux-cuda111 wheel_docker_image: pytorch/manylinux-cuda113
- binary_wheel_upload: - binary_wheel_upload:
context: org-member context: org-member
filters: filters:
...@@ -2359,22 +2214,22 @@ workflows: ...@@ -2359,22 +2214,22 @@ workflows:
- nightly - nightly
tags: tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_windows_wheel_py3.8_cu111_upload name: nightly_binary_windows_wheel_py3.8_cu113_upload
requires: requires:
- nightly_binary_windows_wheel_py3.8_cu111 - nightly_binary_windows_wheel_py3.8_cu113
subfolder: cu111/ subfolder: cu113/
- smoke_test_windows_pip: - smoke_test_windows_pip:
cuda_version: cu111 cuda_version: cu113
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_cu111_smoke_test_pip name: nightly_binary_windows_wheel_py3.8_cu113_smoke_test_pip
python_version: '3.8' python_version: '3.8'
requires: requires:
- nightly_binary_windows_wheel_py3.8_cu111_upload - nightly_binary_windows_wheel_py3.8_cu113_upload
- binary_windows_wheel: - binary_windows_wheel:
cuda_version: cpu cuda_version: cpu
filters: filters:
...@@ -2410,51 +2265,16 @@ workflows: ...@@ -2410,51 +2265,16 @@ workflows:
requires: requires:
- nightly_binary_windows_wheel_py3.9_cpu_upload - nightly_binary_windows_wheel_py3.9_cpu_upload
- binary_windows_wheel: - binary_windows_wheel:
cuda_version: cu102 cuda_version: cu113
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_windows_wheel_py3.9_cu102
python_version: '3.9'
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_windows_wheel_py3.9_cu102_upload
requires:
- nightly_binary_windows_wheel_py3.9_cu102
subfolder: cu102/
- smoke_test_windows_pip:
cuda_version: cu102
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_windows_wheel_py3.9_cu102_smoke_test_pip
python_version: '3.9'
requires:
- nightly_binary_windows_wheel_py3.9_cu102_upload
- binary_windows_wheel:
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.9_cu111 name: nightly_binary_windows_wheel_py3.9_cu113
python_version: '3.9' python_version: '3.9'
wheel_docker_image: pytorch/manylinux-cuda111 wheel_docker_image: pytorch/manylinux-cuda113
- binary_wheel_upload: - binary_wheel_upload:
context: org-member context: org-member
filters: filters:
...@@ -2463,22 +2283,22 @@ workflows: ...@@ -2463,22 +2283,22 @@ workflows:
- nightly - nightly
tags: tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_windows_wheel_py3.9_cu111_upload name: nightly_binary_windows_wheel_py3.9_cu113_upload
requires: requires:
- nightly_binary_windows_wheel_py3.9_cu111 - nightly_binary_windows_wheel_py3.9_cu113
subfolder: cu111/ subfolder: cu113/
- smoke_test_windows_pip: - smoke_test_windows_pip:
cuda_version: cu111 cuda_version: cu113
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.9_cu111_smoke_test_pip name: nightly_binary_windows_wheel_py3.9_cu113_smoke_test_pip
python_version: '3.9' python_version: '3.9'
requires: requires:
- nightly_binary_windows_wheel_py3.9_cu111_upload - nightly_binary_windows_wheel_py3.9_cu113_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
...@@ -3186,49 +3006,15 @@ workflows: ...@@ -3186,49 +3006,15 @@ workflows:
requires: requires:
- nightly_binary_windows_conda_py3.6_cpu_upload - nightly_binary_windows_conda_py3.6_cpu_upload
- binary_windows_conda: - binary_windows_conda:
conda_docker_image: pytorch/conda-builder:cuda102 conda_docker_image: pytorch/conda-builder:cuda113
cuda_version: cu102 cuda_version: cu113
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_windows_conda_py3.6_cu102
python_version: '3.6'
- 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.6_cu102_upload
requires:
- nightly_binary_windows_conda_py3.6_cu102
- smoke_test_windows_conda:
cuda_version: cu102
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_windows_conda_py3.6_cu102_smoke_test_conda
python_version: '3.6'
requires:
- nightly_binary_windows_conda_py3.6_cu102_upload
- binary_windows_conda:
conda_docker_image: pytorch/conda-builder:cuda111
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_conda_py3.6_cu111 name: nightly_binary_windows_conda_py3.6_cu113
python_version: '3.6' python_version: '3.6'
- binary_conda_upload: - binary_conda_upload:
context: org-member context: org-member
...@@ -3238,21 +3024,21 @@ workflows: ...@@ -3238,21 +3024,21 @@ 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_conda_py3.6_cu111_upload name: nightly_binary_windows_conda_py3.6_cu113_upload
requires: requires:
- nightly_binary_windows_conda_py3.6_cu111 - nightly_binary_windows_conda_py3.6_cu113
- smoke_test_windows_conda: - smoke_test_windows_conda:
cuda_version: cu111 cuda_version: cu113
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_conda_py3.6_cu111_smoke_test_conda name: nightly_binary_windows_conda_py3.6_cu113_smoke_test_conda
python_version: '3.6' python_version: '3.6'
requires: requires:
- nightly_binary_windows_conda_py3.6_cu111_upload - nightly_binary_windows_conda_py3.6_cu113_upload
- 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
...@@ -3288,49 +3074,15 @@ workflows: ...@@ -3288,49 +3074,15 @@ workflows:
requires: requires:
- nightly_binary_windows_conda_py3.7_cpu_upload - nightly_binary_windows_conda_py3.7_cpu_upload
- binary_windows_conda: - binary_windows_conda:
conda_docker_image: pytorch/conda-builder:cuda102 conda_docker_image: pytorch/conda-builder:cuda113
cuda_version: cu102 cuda_version: cu113
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_windows_conda_py3.7_cu102
python_version: '3.7'
- 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.7_cu102_upload
requires:
- nightly_binary_windows_conda_py3.7_cu102
- smoke_test_windows_conda:
cuda_version: cu102
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_windows_conda_py3.7_cu102_smoke_test_conda
python_version: '3.7'
requires:
- nightly_binary_windows_conda_py3.7_cu102_upload
- binary_windows_conda:
conda_docker_image: pytorch/conda-builder:cuda111
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_conda_py3.7_cu111 name: nightly_binary_windows_conda_py3.7_cu113
python_version: '3.7' python_version: '3.7'
- binary_conda_upload: - binary_conda_upload:
context: org-member context: org-member
...@@ -3340,21 +3092,21 @@ workflows: ...@@ -3340,21 +3092,21 @@ 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_conda_py3.7_cu111_upload name: nightly_binary_windows_conda_py3.7_cu113_upload
requires: requires:
- nightly_binary_windows_conda_py3.7_cu111 - nightly_binary_windows_conda_py3.7_cu113
- smoke_test_windows_conda: - smoke_test_windows_conda:
cuda_version: cu111 cuda_version: cu113
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_conda_py3.7_cu111_smoke_test_conda name: nightly_binary_windows_conda_py3.7_cu113_smoke_test_conda
python_version: '3.7' python_version: '3.7'
requires: requires:
- nightly_binary_windows_conda_py3.7_cu111_upload - nightly_binary_windows_conda_py3.7_cu113_upload
- 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
...@@ -3390,49 +3142,15 @@ workflows: ...@@ -3390,49 +3142,15 @@ workflows:
requires: requires:
- nightly_binary_windows_conda_py3.8_cpu_upload - nightly_binary_windows_conda_py3.8_cpu_upload
- binary_windows_conda: - binary_windows_conda:
conda_docker_image: pytorch/conda-builder:cuda102 conda_docker_image: pytorch/conda-builder:cuda113
cuda_version: cu102 cuda_version: cu113
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_windows_conda_py3.8_cu102
python_version: '3.8'
- 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.8_cu102_upload
requires:
- nightly_binary_windows_conda_py3.8_cu102
- smoke_test_windows_conda:
cuda_version: cu102
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_windows_conda_py3.8_cu102_smoke_test_conda
python_version: '3.8'
requires:
- nightly_binary_windows_conda_py3.8_cu102_upload
- binary_windows_conda:
conda_docker_image: pytorch/conda-builder:cuda111
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_conda_py3.8_cu111 name: nightly_binary_windows_conda_py3.8_cu113
python_version: '3.8' python_version: '3.8'
- binary_conda_upload: - binary_conda_upload:
context: org-member context: org-member
...@@ -3442,21 +3160,21 @@ workflows: ...@@ -3442,21 +3160,21 @@ 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_conda_py3.8_cu111_upload name: nightly_binary_windows_conda_py3.8_cu113_upload
requires: requires:
- nightly_binary_windows_conda_py3.8_cu111 - nightly_binary_windows_conda_py3.8_cu113
- smoke_test_windows_conda: - smoke_test_windows_conda:
cuda_version: cu111 cuda_version: cu113
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_conda_py3.8_cu111_smoke_test_conda name: nightly_binary_windows_conda_py3.8_cu113_smoke_test_conda
python_version: '3.8' python_version: '3.8'
requires: requires:
- nightly_binary_windows_conda_py3.8_cu111_upload - nightly_binary_windows_conda_py3.8_cu113_upload
- 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
...@@ -3492,49 +3210,15 @@ workflows: ...@@ -3492,49 +3210,15 @@ workflows:
requires: requires:
- nightly_binary_windows_conda_py3.9_cpu_upload - nightly_binary_windows_conda_py3.9_cpu_upload
- binary_windows_conda: - binary_windows_conda:
conda_docker_image: pytorch/conda-builder:cuda102 conda_docker_image: pytorch/conda-builder:cuda113
cuda_version: cu102 cuda_version: cu113
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_windows_conda_py3.9_cu102
python_version: '3.9'
- 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.9_cu102_upload
requires:
- nightly_binary_windows_conda_py3.9_cu102
- smoke_test_windows_conda:
cuda_version: cu102
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_windows_conda_py3.9_cu102_smoke_test_conda
python_version: '3.9'
requires:
- nightly_binary_windows_conda_py3.9_cu102_upload
- binary_windows_conda:
conda_docker_image: pytorch/conda-builder:cuda111
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_conda_py3.9_cu111 name: nightly_binary_windows_conda_py3.9_cu113
python_version: '3.9' python_version: '3.9'
- binary_conda_upload: - binary_conda_upload:
context: org-member context: org-member
...@@ -3544,21 +3228,21 @@ workflows: ...@@ -3544,21 +3228,21 @@ 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_conda_py3.9_cu111_upload name: nightly_binary_windows_conda_py3.9_cu113_upload
requires: requires:
- nightly_binary_windows_conda_py3.9_cu111 - nightly_binary_windows_conda_py3.9_cu113
- smoke_test_windows_conda: - smoke_test_windows_conda:
cuda_version: cu111 cuda_version: cu113
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_conda_py3.9_cu111_smoke_test_conda name: nightly_binary_windows_conda_py3.9_cu113_smoke_test_conda
python_version: '3.9' python_version: '3.9'
requires: requires:
- nightly_binary_windows_conda_py3.9_cu111_upload - nightly_binary_windows_conda_py3.9_cu113_upload
docker_build: docker_build:
triggers: triggers:
- schedule: - schedule:
......
...@@ -22,7 +22,7 @@ import os.path ...@@ -22,7 +22,7 @@ import os.path
PYTHON_VERSIONS = ["3.6", "3.7", "3.8", "3.9"] PYTHON_VERSIONS = ["3.6", "3.7", "3.8", "3.9"]
CU_VERSIONS_DICT = {"linux": ["cpu", "cu102", "cu111","cu113", "rocm4.1"], CU_VERSIONS_DICT = {"linux": ["cpu", "cu102", "cu111","cu113", "rocm4.1"],
"windows": ["cpu", "cu102", "cu111"], "windows": ["cpu", "cu113"],
"macos": ["cpu"]} "macos": ["cpu"]}
...@@ -209,7 +209,7 @@ def unittest_workflows(indentation=6): ...@@ -209,7 +209,7 @@ def unittest_workflows(indentation=6):
job = { job = {
"name": f"unittest_{os_type}_{device_type}_py{python_version}", "name": f"unittest_{os_type}_{device_type}_py{python_version}",
"python_version": python_version, "python_version": python_version,
"cuda_version": 'cpu' if device_type=="cpu" else "cu102", "cuda_version": 'cpu' if device_type=="cpu" else "cu113",
} }
if os_type != "windows": if os_type != "windows":
...@@ -222,7 +222,7 @@ def unittest_workflows(indentation=6): ...@@ -222,7 +222,7 @@ def unittest_workflows(indentation=6):
"stylecheck": { "stylecheck": {
"name": f"stylecheck_py{python_version}", "name": f"stylecheck_py{python_version}",
"python_version": python_version, "python_version": python_version,
"cuda_version": 'cpu' if device_type=="cpu" else "cu102", "cuda_version": 'cpu' if device_type=="cpu" else "cu113",
} }
}) })
return indent(indentation, jobs) return indent(indentation, jobs)
......
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