Release #2340
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| name: Release | |
| on: | |
| push: | |
| branches: | |
| - "release/*" | |
| schedule: | |
| # beijing: 2 am. | |
| # utc: 6 pm. | |
| - cron: "0 18 * * *" | |
| workflow_dispatch: | |
| inputs: | |
| placeholder: | |
| description: "update .github/workflows/release.yml to config your build" | |
| required: false | |
| workflow_call: | |
| inputs: | |
| is_priv: | |
| required: true | |
| type: boolean | |
| branch: | |
| required: false | |
| type: string | |
| default: "main" | |
| upload_override_branch: | |
| required: false | |
| type: string | |
| cuda_cmake_cache: | |
| required: false | |
| type: string | |
| secrets: | |
| ONEFLOW_PRIV_ORG: | |
| required: true | |
| ONEFLOW_PRIV_GH_TOKEN: | |
| required: true | |
| ONEFLOW_PRIV_OSS_BUCKET: | |
| required: true | |
| OSS_ACCESS_KEY_ID: | |
| required: true | |
| OSS_ACCESS_KEY_SECRET: | |
| required: true | |
| ONEFLOW_CI_HTTP_PROXY: | |
| required: false | |
| concurrency: | |
| group: release-${{ github.ref }}-${{ inputs.branch }} | |
| cancel-in-progress: ${{ github.ref != 'refs/heads/master' }} | |
| env: | |
| ONEFLOW_SRC: . | |
| jobs: | |
| generate-build-matrix: | |
| name: "Generate build matrix" | |
| runs-on: ubuntu-latest | |
| env: | |
| ONEFLOW_SRC: . | |
| outputs: | |
| matrix: ${{ steps.find-cache.outputs.matrix }} | |
| formatted_date: ${{ steps.date.outputs.formatted_date }} | |
| steps: | |
| - name: Checkout Oneflow-Inc/oneflow | |
| uses: actions/checkout@v2 | |
| if: ${{ !inputs.is_priv }} | |
| with: | |
| ref: ${{ github.event.pull_request.head.sha }} | |
| repository: ${{github.event.pull_request.head.repo.full_name}} | |
| - name: Checkout oneflow | |
| uses: actions/checkout@v2 | |
| if: ${{ inputs.is_priv }} | |
| with: | |
| ref: ${{ inputs.branch }} | |
| repository: ${{ secrets.ONEFLOW_PRIV_ORG }}/oneflow | |
| token: ${{ secrets.ONEFLOW_PRIV_GH_TOKEN }} | |
| - uses: Oneflow-Inc/get-oneflow/cache-complete/matrix/build@whl-skip-nccl | |
| name: Find build cache | |
| id: find-cache | |
| timeout-minutes: 5 | |
| with: | |
| delete-cache: ${{ contains(github.event.pull_request.labels.*.name, 'need-clean-ccache') }} | |
| runner-labels: | | |
| self-hosted | |
| linux | |
| release | |
| oneflow-src: ${{ env.ONEFLOW_SRC }} | |
| entries: | | |
| cu122 | |
| cu121 | |
| cu118 | |
| cpu | |
| - name: Get current date | |
| id: date | |
| run: echo "formatted_date=$(date +'%Y%m%d')" >> $GITHUB_OUTPUT | |
| staging_release: | |
| env: | |
| MANYLINUX_CACHE_DIR: ~/manylinux-cache-dir/release/${{ matrix.entry }} | |
| WHEELHOUSE_DIR: manylinux_wheelhouse | |
| OSS_DIR: branch/${{ github.ref_name }}/${{ matrix.entry }}/${{ github.sha }} | |
| GITHUB_REF_NAME: ${{ github.ref_name }} | |
| GITHUB_SHA: ${{ github.sha }} | |
| ONEFLOW_OSS_BUCKET: oneflow-staging | |
| https_proxy: ${{ secrets.ONEFLOW_CI_HTTP_PROXY }} | |
| needs: [generate-build-matrix] | |
| name: Staging Release | |
| timeout-minutes: 240 | |
| runs-on: [self-hosted, linux, release] | |
| if: github.repository == 'Oneflow-Inc/oneflow' || inputs.is_priv | |
| strategy: | |
| fail-fast: false | |
| max-parallel: 6 | |
| matrix: ${{ fromJson(needs.generate-build-matrix.outputs.matrix) }} | |
| steps: | |
| - name: Fix permissions | |
| run: | | |
| docker run --rm -v $PWD:/p -w /p busybox rm -rf * | |
| - name: Install dependencies | |
| run: | | |
| python3 -m pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple | |
| python3 -m pip install -U setuptools wheel --user | |
| python3 -m pip install oss2 --user | |
| - name: Checkout Oneflow-Inc/oneflow | |
| uses: actions/checkout@v2 | |
| if: ${{ !inputs.is_priv }} | |
| with: | |
| ref: ${{ github.event.pull_request.head.sha }} | |
| repository: ${{github.event.pull_request.head.repo.full_name}} | |
| - name: Checkout private oneflow | |
| uses: actions/checkout@v2 | |
| if: ${{ inputs.is_priv }} | |
| with: | |
| ref: ${{ inputs.branch }} | |
| repository: ${{ secrets.ONEFLOW_PRIV_ORG }}/oneflow | |
| token: ${{ secrets.ONEFLOW_PRIV_GH_TOKEN }} | |
| - name: Checkout cutlass_extension | |
| uses: actions/checkout@v2 | |
| if: ${{ inputs.is_priv }} | |
| with: | |
| repository: ${{ secrets.ONEFLOW_PRIV_ORG }}/cutlass-extension | |
| token: ${{ secrets.ONEFLOW_PRIV_GH_TOKEN }} | |
| path: cutlass-extension | |
| - name: Set Private env | |
| if: ${{ inputs.is_priv }} | |
| run: | | |
| GITHUB_SHA=$(git rev-parse HEAD) | |
| echo "OSS_DIR=branch/${{ inputs.upload_override_branch || inputs.branch }}/${{ matrix.entry }}/${GITHUB_SHA}" >> $GITHUB_ENV | |
| echo "GITHUB_REF_NAME=${{ inputs.upload_override_branch || inputs.branch }}" >> $GITHUB_ENV | |
| echo "GITHUB_SHA=${GITHUB_SHA}" >> $GITHUB_ENV | |
| echo "ONEFLOW_OSS_BUCKET=${{ secrets.ONEFLOW_PRIV_OSS_BUCKET }}" >> $GITHUB_ENV | |
| - name: Print env | |
| if: ${{ inputs.is_priv }} | |
| run: | | |
| env | |
| - uses: Oneflow-Inc/get-oneflow@whl-skip-nccl | |
| name: Build ${{ matrix.entry }} | |
| if: ${{ matrix.entry =='cu118' || startsWith(matrix.entry, 'cu12') }} | |
| with: | |
| cmake-init-cache: ${{ env.ONEFLOW_SRC }}/${{ inputs.cuda_cmake_cache || 'cmake/caches/ci/release/cu118.cmake' }} | |
| build-script: ${{ env.ONEFLOW_SRC }}/ci/manylinux/build-gcc9.sh | |
| oneflow-src: ${{ env.ONEFLOW_SRC }} | |
| oneflow-build-env: manylinux | |
| wheelhouse-dir: ${{ env.WHEELHOUSE_DIR }} | |
| clear-wheelhouse-dir: true | |
| self-hosted: true | |
| compute-platform: ${{ matrix.entry }} | |
| manylinux-cache-dir: ${{ env.MANYLINUX_CACHE_DIR }} | |
| docker-run-use-system-http-proxy: false | |
| docker-run-use-lld: false | |
| retry-failed-build: true | |
| clean-ccache: true | |
| nightly: ${{ inputs.is_priv || github.event_name == 'schedule' || github.ref == 'refs/heads/release/add_nightly_date_index'}} | |
| nightly-date: ${{ needs.generate-build-matrix.outputs.formatted_date }} | |
| use-nvidia-wheels: ${{ matrix.entry !='cu112' }} | |
| python-versions: | | |
| 3.11 | |
| 3.10 | |
| 3.9 | |
| 3.8 | |
| - uses: Oneflow-Inc/get-oneflow@whl-skip-nccl | |
| name: Build ${{ matrix.entry }} | |
| if: ${{ startsWith(matrix.entry, 'cu') && matrix.entry !='cu118' && !startsWith(matrix.entry, 'cu12') }} | |
| with: | |
| cmake-init-cache: ${{ env.ONEFLOW_SRC }}/cmake/caches/ci/release/cuda.cmake | |
| build-script: ${{ env.ONEFLOW_SRC }}/ci/manylinux/build-gcc9.sh | |
| oneflow-src: ${{ env.ONEFLOW_SRC }} | |
| oneflow-build-env: manylinux | |
| wheelhouse-dir: ${{ env.WHEELHOUSE_DIR }} | |
| clear-wheelhouse-dir: true | |
| self-hosted: true | |
| compute-platform: ${{ matrix.entry }} | |
| manylinux-cache-dir: ${{ env.MANYLINUX_CACHE_DIR }} | |
| docker-run-use-system-http-proxy: false | |
| docker-run-use-lld: false | |
| retry-failed-build: true | |
| clean-ccache: true | |
| nightly: ${{ inputs.is_priv || github.event_name == 'schedule' || github.ref == 'refs/heads/release/add_nightly_date_index'}} | |
| nightly-date: ${{ needs.generate-build-matrix.outputs.formatted_date }} | |
| use-nvidia-wheels: ${{ matrix.entry !='cu112' }} | |
| python-versions: | | |
| 3.11 | |
| 3.10 | |
| 3.9 | |
| 3.8 | |
| - uses: Oneflow-Inc/get-oneflow@whl-skip-nccl | |
| name: Build ${{ matrix.entry }} | |
| if: ${{ matrix.entry =='cpu' }} | |
| with: | |
| cmake-init-cache: ${{ env.ONEFLOW_SRC }}/cmake/caches/ci/release/cpu.cmake | |
| build-script: ${{ env.ONEFLOW_SRC }}/ci/manylinux/build.sh | |
| oneflow-src: ${{ env.ONEFLOW_SRC }} | |
| oneflow-build-env: manylinux | |
| wheelhouse-dir: ${{ env.WHEELHOUSE_DIR }} | |
| clear-wheelhouse-dir: true | |
| self-hosted: true | |
| compute-platform: ${{ matrix.entry }} | |
| manylinux-cache-dir: ${{ env.MANYLINUX_CACHE_DIR }} | |
| docker-run-use-system-http-proxy: false | |
| docker-run-use-lld: false | |
| retry-failed-build: true | |
| clean-ccache: false | |
| nightly: ${{ inputs.is_priv || github.event_name == 'schedule' || github.ref == 'refs/heads/release/add_nightly_date_index'}} | |
| nightly-date: ${{ needs.generate-build-matrix.outputs.formatted_date }} | |
| python-versions: | | |
| 3.11 | |
| 3.10 | |
| 3.9 | |
| 3.8 | |
| - name: Upload wheel | |
| uses: ./.github/actions/upload_oss | |
| with: | |
| src_path: ${{ env.WHEELHOUSE_DIR }} | |
| oss_dst_path: oss://${{ env.ONEFLOW_OSS_BUCKET }}/${{ env.OSS_DIR }} | |
| oss_access_key_id: ${{ secrets.OSS_ACCESS_KEY_ID }} | |
| oss_access_key_secret: ${{ secrets.OSS_ACCESS_KEY_SECRET }} | |
| - name: Update pip index | |
| env: | |
| OSS_ACCESS_KEY_ID: ${{ secrets.OSS_ACCESS_KEY_ID }} | |
| OSS_ACCESS_KEY_SECRET: ${{ secrets.OSS_ACCESS_KEY_SECRET }} | |
| run: | | |
| python3 -m pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple | |
| python3 -m pip install oss2 beautifulsoup4 --user | |
| python3 tools/create_pip_index.py --dir_key ${{ env.OSS_DIR }} -b ${{ env.ONEFLOW_OSS_BUCKET }} \ | |
| --index_key=branch/${{ env.GITHUB_REF_NAME }}/${{ matrix.entry }}/index.html \ | |
| --index_key=branch/${{ env.GITHUB_REF_NAME }}/date/${{ needs.generate-build-matrix.outputs.formatted_date }}/${{ matrix.entry }}/index.html \ | |
| --index_key=${{ env.OSS_DIR }}/index.html \ | |
| --index_key=commit/${{ env.GITHUB_SHA }}/${{ matrix.entry }}/index.html | |
| - name: Update API docs | |
| if: github.ref == 'refs/heads/master' && matrix.entry == 'cpu' && !inputs.is_priv | |
| env: | |
| READTHEDOCS_TOKEN: ${{ secrets.READTHEDOCS_TOKEN }} | |
| run: | | |
| curl -X POST -d "branches=master" -d "token=${READTHEDOCS_TOKEN}" https://readthedocs.org/api/v2/webhook/oneflow/135376/ |