stable-fast/.github/workflows/wheels.yml at 616944ee261b189f26a53d75e3e0a243fa095127 · PrunaAI/stable-fast · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
name: wheels

on:
  workflow_dispatch:

jobs:
  build-local:
    strategy:
      fail-fast: false
      matrix:
        os:
          - ubuntu-22.04
          # - windows-2019
        python: ['3.10', '3.11', '3.12']
        torch_version: ['2.10.0']
        cuda_short_version: ['126']

    uses: ./.github/workflows/wheels_build.yml
    with:
      os: ${{ matrix.os }}
      python: ${{ matrix.python }}
      torch_version: ${{ matrix.torch_version }}
      cuda_short_version: ${{ matrix.cuda_short_version }}

  build-pypi:
    # Single canonical build intended for PyPI: no local CUDA/torch suffix
    strategy:
      fail-fast: false
      matrix:
        os: ['ubuntu-22.04']
        python: ['3.10', '3.11', '3.12']

    uses: ./.github/workflows/wheels_build.yml
    with:
      os: ${{ matrix.os }}
      python: ${{ matrix.python }}
      torch_version: '2.10.0'
      cuda_short_version: '128'
      append_local_version: '0'  # 0 to disable local version suffix

  # publish to GitHub Release
  # gh_release:
  #   name: gh_release
  #   needs: build
  #   runs-on: ubuntu-20.04

  #   timeout-minutes: 360
  #   defaults:
  #     run:
  #       shell: bash
  #   steps:
  #     - uses: actions/download-artifact@v4
  #       with:
  #         path: dist

  #     - run: ls -R dist/

  #     # create night release if it's a push to main
  #     - if: github.event_name == 'push' && github.ref == 'refs/heads/main'
  #       name: Nightly Release
  #       uses: andelf/nightly-release@v1
  #       env:
  #         GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
  #       with:
  #         tag_name: nightly
  #         name: 'stable-fast Nightly Release $$'
  #         prerelease: true
  #         body: 'TODO: Add nightly release notes'
  #         files: |
  #           dist/*/*.whl

  #     # create release if it's a tag like vx.y.z
  #     - if: github.ref_type == 'tag' && startsWith(github.ref, 'refs/tags/v')
  #       name: Release
  #       uses: softprops/action-gh-release@v1
  #       with :
  #         files: |
  #           dist/*/*.whl


  consolidate-wheels:
    needs: [build-local, build-pypi]
    runs-on: ubuntu-latest
    steps:
      - name: Download all wheel artifacts
        uses: actions/download-artifact@v4
        with:
          path: dist

      - name: Consolidate wheels into a single folder
        run: |
          mkdir -p consolidated_wheels
          find dist -name '*.whl' -exec cp {} consolidated_wheels/ \;
          ls -l consolidated_wheels

      - name: Upload consolidated wheels
        uses: actions/upload-artifact@v4
        with:
          name: built-wheels
          path: consolidated_wheels

  # upload_pip:
  #   needs: build
  #   uses: ./.github/workflows/wheels_upload_pip.yml
  #   with:
  #     twine_username: __token__
  #     filter: "*torch2.1.0+cu121*"
  #     execute: ${{ github.repository == 'chengzeyi/stable-fast' && github.event_name != 'pull_request' }}
  #   secrets:
  #     twine_password: ${{ secrets.PYPI_TOKEN }}

  # upload_pt_cu118:
  #   needs: build
  #   uses: ./.github/workflows/wheels_upload_s3.yml
  #   with:
  #     aws_role: "arn:aws:iam::749337293305:role/pytorch_bot_uploader_role"
  #     s3_path: s3://pytorch/whl/cu118/
  #     aws_s3_cp_extra_args: --acl public-read
  #     filter: "*torch2.1.0+cu118*"
  #     execute: ${{ github.repository == 'chengzeyi/stable-fast' && github.ref_type == 'tag' }}

  # upload_pt_cu121:
  #   needs: build
  #   uses: ./.github/workflows/wheels_upload_s3.yml
  #   with:
  #     aws_role: "arn:aws:iam::749337293305:role/pytorch_bot_uploader_role"
  #     s3_path: s3://pytorch/whl/cu121/
  #     aws_s3_cp_extra_args: --acl public-read
  #     filter: "*torch2.1.0+cu121*"
  #     execute: ${{ github.repository == 'chengzeyi/stable-fast' && github.ref_type == 'tag' }}