_publish.yaml

# Copyright (c) Meta Platforms, Inc. and affiliates.
# All rights reserved.
#
# This source code is licensed under the BSD-style license found in the
# LICENSE file in the root directory of this source tree.

on:
  workflow_call:
    inputs:
      release_type:
        type: string
        required: true

env:
    ACTIONS_ALLOW_USE_UNSECURE_NODE_VERSION: true

jobs:
  publish_pt21_s3-linux:
    name: Publish to S3 (pt${{ matrix.torch }}, py${{ matrix.py }}, linux-x86_64, ${{ matrix.variant }})
    strategy:
      matrix:
        torch: ['2.1.0', '2.1.1', '2.1.2']
        py: ['3.8', '3.9', '3.10', '3.11']
        variant: ['cpu', 'cu118', 'cu121']
      max-parallel: 1
    uses: ./.github/workflows/_publish_s3.yaml
    with:
      os: 'linux'
      torch: ${{ matrix.torch }}
      py: ${{ matrix.py }}
      variant: ${{ matrix.variant }}
      release_type: ${{ inputs.release_type }}

  publish_pt22_s3-linux:
    name: Publish to S3 (pt${{ matrix.torch }}, py${{ matrix.py }}, linux-x86_64, ${{ matrix.variant }})
    strategy:
      matrix:
        torch: ['2.2.0', '2.2.1', '2.2.2']
        py: ['3.8', '3.9', '3.10', '3.11']
        variant: ['cpu', 'cu118', 'cu121']
      max-parallel: 1
    uses: ./.github/workflows/_publish_s3.yaml
    with:
      os: 'linux'
      torch: ${{ matrix.torch }}
      py: ${{ matrix.py }}
      variant: ${{ matrix.variant }}
      release_type: ${{ inputs.release_type }}

  publish_pt23_s3-linux:
    name: Publish to S3 (pt${{ matrix.torch }}, py${{ matrix.py }}, linux-x86_64, ${{ matrix.variant }})
    strategy:
      matrix:
        torch: ['2.3.0']
        py: ['3.8', '3.9', '3.10', '3.11']
        variant: ['cpu', 'cu118', 'cu121']
      max-parallel: 1
    uses: ./.github/workflows/_publish_s3.yaml
    with:
      os: 'linux'
      torch: ${{ matrix.torch }}
      py: ${{ matrix.py }}
      variant: ${{ matrix.variant }}
      release_type: ${{ inputs.release_type }}

  publish_pt23_s3-macos:
    name: Publish to S3 (pt${{ matrix.torch }}, py${{ matrix.py }}, macos-arm64)
    strategy:
      matrix:
        torch: ['2.3.0']
        py: ['3.8', '3.9', '3.10', '3.11']
      max-parallel: 1
    uses: ./.github/workflows/_publish_s3.yaml
    with:
      os: 'macos'
      torch: ${{ matrix.torch }}
      py: ${{ matrix.py }}
      variant: 'cpu'
      release_type: ${{ inputs.release_type }}
      arch: 'arm64'

  publish_pypi-linux:
    name: Publish to PyPI (pt${{ matrix.torch }}, py${{ matrix.py }}, linux-x86_64, ${{ matrix.variant }})
    needs:
      - publish_pt21_s3-linux
      - publish_pt22_s3-linux
      - publish_pt23_s3-linux
      - publish_pt23_s3-macos
    if: inputs.release_type == 'stable'
    uses: ./.github/workflows/_publish_pypi.yaml
    strategy:
      matrix:
        torch: ['2.3.0']
        py: ['3.8', '3.9', '3.10', '3.11']
        variant: ['cu121']
      max-parallel: 1
    with:
      os: 'linux'
      torch: ${{ matrix.torch }}
      py: ${{ matrix.py }}
      variant: ${{ matrix.variant }}

  publish_pypi-macos:
    name: Publish to PyPI (pt${{ matrix.torch }}, py${{ matrix.py }}, macos-arm64)
    needs:
      - publish_pypi-linux
    if: inputs.release_type == 'stable'
    uses: ./.github/workflows/_publish_pypi.yaml
    strategy:
      matrix:
        torch: ['2.3.0']
        py: ['3.8', '3.9', '3.10', '3.11']
      max-parallel: 1
    with:
      os: 'macos'
      torch: ${{ matrix.torch }}
      py: ${{ matrix.py }}
      arch: 'arm64'

  publish_doc:
    name: Publish documentation
    needs:
      - publish_pypi-linux
      - publish_pypi-macos
    # We can't use `success()` here, see
    # https://github.com/orgs/community/discussions/45058.
    if: ${{ !failure() && !cancelled() }}
    uses: ./.github/workflows/_publish_doc.yaml