diff --git a/.circleci/config.yml b/.circleci/config.yml index be43aa12a81..adc99f5a68a 100644 --- a/.circleci/config.yml +++ b/.circleci/config.yml @@ -1205,6 +1205,11 @@ workflows: python_version: '3.9' - binary_win_wheel: cu_version: cpu + filters: + branches: + only: main + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ name: binary_win_wheel_py3.10_cpu python_version: '3.10' - binary_win_wheel: @@ -1227,8 +1232,39 @@ workflows: python_version: '3.10' - binary_win_wheel: cu_version: cu118 + filters: + branches: + only: main + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ name: binary_win_wheel_py3.10_cu118 python_version: '3.10' + - binary_win_wheel: + cu_version: cpu + name: binary_win_wheel_py3.11_cpu + python_version: '3.11' + - binary_win_wheel: + cu_version: cu116 + filters: + branches: + only: main + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: binary_win_wheel_py3.11_cu116 + python_version: '3.11' + - binary_win_wheel: + cu_version: cu117 + filters: + branches: + only: main + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: binary_win_wheel_py3.11_cu117 + python_version: '3.11' + - binary_win_wheel: + cu_version: cu118 + name: binary_win_wheel_py3.11_cu118 + python_version: '3.11' - binary_win_conda: cu_version: cpu filters: @@ -1303,6 +1339,11 @@ workflows: python_version: '3.9' - binary_win_conda: cu_version: cpu + filters: + branches: + only: main + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ name: binary_win_conda_py3.10_cpu python_version: '3.10' - binary_win_conda: @@ -1325,8 +1366,39 @@ workflows: python_version: '3.10' - binary_win_conda: cu_version: cu118 + filters: + branches: + only: main + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ name: binary_win_conda_py3.10_cu118 python_version: '3.10' + - binary_win_conda: + cu_version: cpu + name: binary_win_conda_py3.11_cpu + python_version: '3.11' + - binary_win_conda: + cu_version: cu116 + filters: + branches: + only: main + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: binary_win_conda_py3.11_cu116 + python_version: '3.11' + - binary_win_conda: + cu_version: cu117 + filters: + branches: + only: main + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: binary_win_conda_py3.11_cu117 + python_version: '3.11' + - binary_win_conda: + cu_version: cu118 + name: binary_win_conda_py3.11_cu118 + python_version: '3.11' - build_docs: filters: branches: @@ -1391,6 +1463,15 @@ workflows: - nightly name: unittest_linux_gpu_py3.10 python_version: '3.10' + - unittest_linux_gpu: + cu_version: cu116 + filters: + branches: + only: + - main + - nightly + name: unittest_linux_gpu_py3.11 + python_version: '3.11' - unittest_windows_cpu: cu_version: cpu name: unittest_windows_cpu_py3.8 @@ -1403,6 +1484,10 @@ workflows: cu_version: cpu name: unittest_windows_cpu_py3.10 python_version: '3.10' + - unittest_windows_cpu: + cu_version: cpu + name: unittest_windows_cpu_py3.11 + python_version: '3.11' - unittest_windows_gpu: cu_version: cu116 name: unittest_windows_gpu_py3.8 @@ -1425,6 +1510,15 @@ workflows: - nightly name: unittest_windows_gpu_py3.10 python_version: '3.10' + - unittest_windows_gpu: + cu_version: cu116 + filters: + branches: + only: + - main + - nightly + name: unittest_windows_gpu_py3.11 + python_version: '3.11' - unittest_macos_cpu: cu_version: cpu name: unittest_macos_cpu_py3.8 @@ -1437,6 +1531,10 @@ workflows: cu_version: cpu name: unittest_macos_cpu_py3.10 python_version: '3.10' + - unittest_macos_cpu: + cu_version: cpu + name: unittest_macos_cpu_py3.11 + python_version: '3.11' cmake: jobs: @@ -1804,6 +1902,86 @@ workflows: requires: - nightly_binary_win_wheel_py3.10_cu118 subfolder: cu118/ + - binary_win_wheel: + cu_version: cpu + filters: + branches: + only: nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_win_wheel_py3.11_cpu + python_version: '3.11' + - binary_wheel_upload: + context: org-member + filters: + branches: + only: nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_win_wheel_py3.11_cpu_upload + requires: + - nightly_binary_win_wheel_py3.11_cpu + subfolder: cpu/ + - binary_win_wheel: + cu_version: cu116 + filters: + branches: + only: nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_win_wheel_py3.11_cu116 + python_version: '3.11' + - binary_wheel_upload: + context: org-member + filters: + branches: + only: nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_win_wheel_py3.11_cu116_upload + requires: + - nightly_binary_win_wheel_py3.11_cu116 + subfolder: cu116/ + - binary_win_wheel: + cu_version: cu117 + filters: + branches: + only: nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_win_wheel_py3.11_cu117 + python_version: '3.11' + - binary_wheel_upload: + context: org-member + filters: + branches: + only: nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_win_wheel_py3.11_cu117_upload + requires: + - nightly_binary_win_wheel_py3.11_cu117 + subfolder: cu117/ + - binary_win_wheel: + cu_version: cu118 + filters: + branches: + only: nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_win_wheel_py3.11_cu118 + python_version: '3.11' + - binary_wheel_upload: + context: org-member + filters: + branches: + only: nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_win_wheel_py3.11_cu118_upload + requires: + - nightly_binary_win_wheel_py3.11_cu118 + subfolder: cu118/ - binary_win_conda: cu_version: cpu filters: @@ -2032,6 +2210,82 @@ workflows: name: nightly_binary_win_conda_py3.10_cu118_upload requires: - nightly_binary_win_conda_py3.10_cu118 + - binary_win_conda: + cu_version: cpu + filters: + branches: + only: nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_win_conda_py3.11_cpu + python_version: '3.11' + - binary_conda_upload: + context: org-member + filters: + branches: + only: nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_win_conda_py3.11_cpu_upload + requires: + - nightly_binary_win_conda_py3.11_cpu + - binary_win_conda: + cu_version: cu116 + filters: + branches: + only: nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_win_conda_py3.11_cu116 + python_version: '3.11' + - binary_conda_upload: + context: org-member + filters: + branches: + only: nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_win_conda_py3.11_cu116_upload + requires: + - nightly_binary_win_conda_py3.11_cu116 + - binary_win_conda: + cu_version: cu117 + filters: + branches: + only: nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_win_conda_py3.11_cu117 + python_version: '3.11' + - binary_conda_upload: + context: org-member + filters: + branches: + only: nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_win_conda_py3.11_cu117_upload + requires: + - nightly_binary_win_conda_py3.11_cu117 + - binary_win_conda: + cu_version: cu118 + filters: + branches: + only: nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_win_conda_py3.11_cu118 + python_version: '3.11' + - binary_conda_upload: + context: org-member + filters: + branches: + only: nightly + tags: + only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ + name: nightly_binary_win_conda_py3.11_cu118_upload + requires: + - nightly_binary_win_conda_py3.11_cu118 docker_build: triggers: - schedule: diff --git a/.circleci/regenerate.py b/.circleci/regenerate.py index 2fee884ef1e..6c9c86e294b 100755 --- a/.circleci/regenerate.py +++ b/.circleci/regenerate.py @@ -21,7 +21,7 @@ from jinja2 import select_autoescape -PYTHON_VERSIONS = ["3.8", "3.9", "3.10"] +PYTHON_VERSIONS = ["3.8", "3.9", "3.10", "3.11"] RC_PATTERN = r"/v[0-9]+(\.[0-9]+)*-rc[0-9]+/" diff --git a/.github/workflows/test-linux-cpu.yml b/.github/workflows/test-linux-cpu.yml index 5dc7550d868..0527f0a26aa 100644 --- a/.github/workflows/test-linux-cpu.yml +++ b/.github/workflows/test-linux-cpu.yml @@ -16,7 +16,7 @@ jobs: tests: strategy: matrix: - python_version: ["3.8", "3.9", "3.10"] + python_version: ["3.8", "3.9", "3.10", "3.11"] fail-fast: false uses: pytorch/test-infra/.github/workflows/linux_job.yml@main with: diff --git a/packaging/pkg_helpers.bash b/packaging/pkg_helpers.bash index 69f73391869..0e8c17d4122 100644 --- a/packaging/pkg_helpers.bash +++ b/packaging/pkg_helpers.bash @@ -150,7 +150,7 @@ retry () { } # Inputs: -# PYTHON_VERSION (3.8, 3.9, 3.10) +# PYTHON_VERSION (3.8, 3.9, 3.10, 3.11) # UNICODE_ABI (bool) # # Outputs: @@ -172,6 +172,7 @@ setup_wheel_python() { 3.8) python_abi=cp38-cp38 ;; 3.9) python_abi=cp39-cp39 ;; 3.10) python_abi=cp310-cp310 ;; + 3.11) python_abi=cp311-cp311 ;; *) echo "Unrecognized PYTHON_VERSION=$PYTHON_VERSION" exit 1