build_linux_wheels.yaml

name: Build Linux Wheels

on:
  push:
    branches:
      - nightly
      - 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]+
  workflow_dispatch:
  pull_request:

permissions:
  id-token: write
  contents: read

jobs:
  generate-matrix:
    if: github.repository_owner == 'pytorch'
    uses: pytorch/test-infra/.github/workflows/generate_binary_build_matrix.yml@main
    with:
      package-type: wheel
      os: linux
      test-infra-repository: pytorch/test-infra
      test-infra-ref: main
      with-cuda: enable
      with-rocm: enable
      build-python-only: enable
  test:
    needs: generate-matrix
    runs-on: ubuntu-latest
    steps:
      - name: one
        shell: python
        env:
          A: ${{ needs.generate-matrix.outputs.matrix }}
        run: |
          import os
          import json
          a = os.environ["A"]
          print(type(a))
          a_as_dict = json.loads(a)
          print(type(a_as_dict))
          # new_dict = {"include": []}
          # for build in a["include"]:
          #   if build["python_version"] != "3.13":
          #     new_dict["include"].append(build)
          # print("hello")
  # build:
  #   if: ${{ needs.generate-matrix.outputs.python_version != '3.9' }}
  #   needs: generate-matrix
  #   name: ${{ matrix.repository }}
  #   uses: pytorch/test-infra/.github/workflows/build_wheels_linux.yml@main
  #   strategy:
  #     fail-fast: false
  #   with:
  #     repository: pytorch/torchtune
  #     ref: ""
  #     package-name: torchtune
  #     build-matrix: ${{ needs.generate-matrix.outputs.matrix }}
  #     pre-script: .github/scripts/pre_build_script.sh
  #     trigger-event: ${{ github.event_name }}
  #     build-platform: 'python-build-package'
  #     pip-install-torch-extra-args:
  #       torchvision
  #       torchao
      # - name: output-python-version
      #   run: echo PYTHON_VERSION=$env.PYTHON_VERSION >> "$GITHUB_OUTPUT"