# From https://github.com/pytorch/test-infra/wiki/Using-Nova-Reusable-Build-Workflows
name: Build Linux Wheels

on:
  pull_request:
    paths:
      - build/packaging/**
      - .github/workflows/build_wheels_linux.yml
      - setup.py
  push:
    branches:
      - nightly
      - main
      - release/*
    tags:
        # NOTE: Binary build pipelines should only get triggered on release candidate builds
        # Release candidate tags look like: v1.11.0-rc1
        - v[0-9]+.[0-9]+.[0-9]+-rc[0-9]+
  schedule:
    - cron: '0 0 * * *'  # Runs at midnight UTC every day
  workflow_dispatch:

jobs:
  generate-matrix:
    uses: pytorch/test-infra/.github/workflows/generate_binary_build_matrix.yml@main
    with:
      package-type: wheel
      os: linux
      with-cpu: enable
      with-cuda: enable
      with-rocm: enable
      with-xpu: enable
      # Note: if free-threaded python is required add py3.13t here
      python-versions: '["3.9"]'

  build:
    needs: generate-matrix
    permissions:
      id-token: write
      contents: read
    uses: pytorch/test-infra/.github/workflows/build_wheels_linux.yml@main
    with:
      # Set the ref to an empty string instead of the default nightly because
      # torchao doesn't have nightly branch setup yet, instead the build is
      # triggered daily from main with a schedule
      repository: pytorch/ao
      ref: ""
      test-infra-repository: pytorch/test-infra
      test-infra-ref: main
      build-matrix: ${{ needs.generate-matrix.outputs.matrix }}
      env-var-script: packaging/env_var_script_linux.sh
      pre-script: packaging/pre_build_script.sh
      post-script: packaging/post_build_script.sh
      smoke-test-script: packaging/smoke_test.py
      package-name: torchao
      trigger-event: ${{ github.event_name }}
      # This is the CUDA version to be uploaded to torchao-nightly pypi
      upload-to-pypi: cu121
    secrets:
      PYPI_API_TOKEN: ${{ secrets.PYPI_API_TOKEN }}
  notify:
    runs-on: ubuntu-latest
    name: Email notification
    needs: [generate-matrix, build]
    if: failure() && github.event_name == 'schedule'
    steps:
      - uses: dawidd6/action-send-mail@v4
        with:
          server_address: smtp.gmail.com
          server_port: 465
          username: torchao.notify
          password: ${{ secrets.TORCHAO_NOTIFY_PASSWORD }}
          from: torchao.notify@gmail.com
          to: ${{ secrets.TORCHAO_NOTIFY_RECIPIENT }}
          subject: Scheduled Build Failure for TorchAO
          body: |
            Build Failure Notification for TorchAO

            A failure occurred in the Build Linux Wheels workflow.

            Run Details:
            - Workflow: ${{ github.workflow }}
            - Run Type: ${{ github.event_name }}
            - Repository: ${{ github.repository }}
            - Branch/PR: ${{ github.ref }}
            - Commit: ${{ github.sha }}

            You can view the full run details here:
            ${{ github.server_url }}/${{ github.repository }}/actions/runs/${{ github.run_id }}

            Error Information:
            ${{ needs.generate-matrix.result == 'failure' && 'Matrix generation failed' || '' }}
            ${{ needs.build.result == 'failure' && 'Build job failed' || '' }}

            This is an automated notification. Please check the GitHub Actions page for more details about the failure.