add_npu_support for goup_norm (#10496) #2450
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: Canary | |
| on: | |
| push: | |
| branches: | |
| - master | |
| - "canary/*" | |
| workflow_dispatch: | |
| inputs: | |
| oneflow-ref: | |
| description: "" | |
| default: "master" | |
| required: true | |
| concurrency: | |
| group: canary-${{ github.ref }} | |
| cancel-in-progress: false | |
| jobs: | |
| canary_release: | |
| name: Canary Release | |
| timeout-minutes: 120 | |
| runs-on: [self-hosted, linux, release] | |
| if: github.repository == 'Oneflow-Inc/oneflow' | |
| strategy: | |
| max-parallel: 1 | |
| fail-fast: false | |
| matrix: | |
| entry: ["canary", "profiler"] | |
| include: | |
| - entry: "canary" | |
| cmake-init-cache: "cmake/caches/ci/canary/cuda.cmake" | |
| - entry: "profiler" | |
| cmake-init-cache: "cmake/caches/ci/profiler/cuda.cmake" | |
| env: | |
| ONEFLOW_SRC: . | |
| MANYLINUX_CACHE_DIR: ~/manylinux-cache-dir/canary-cu112 | |
| WHEELHOUSE_DIR: manylinux-wheelhouse | |
| COMPUTE_PLATFORM: cu117 | |
| OSS_BUCKET: oneflow-staging | |
| OSS_WHEEL_HOUSE_DIR: ${{ matrix.entry }}/commit/${{ github.sha }} | |
| OSS_GITHUB_REF_DIR: ${{ matrix.entry }}/${{ github.ref }} | |
| steps: | |
| - name: Fix permissions | |
| run: | | |
| set -x | |
| docker run --rm -v $PWD:$PWD -w $PWD busybox rm -rf * | |
| - name: Remove leftover cuda-installer.log | |
| run: | | |
| docker run --rm -v /tmp:/host/tmp -w /p busybox rm -f /host/tmp/cuda-installer.log | |
| - name: Checkout Oneflow-Inc/oneflow | |
| if: ${{ github.event.inputs.oneflow-ref != '' }} | |
| uses: actions/checkout@v2 | |
| with: | |
| ref: ${{ github.event.inputs.oneflow-ref }} | |
| - name: Checkout Oneflow-Inc/oneflow | |
| if: ${{ github.event.inputs.oneflow-ref == '' }} | |
| uses: actions/checkout@v2 | |
| - uses: Oneflow-Inc/get-oneflow@whl-skip-nccl | |
| name: Build manylinux | |
| id: build-cuda | |
| with: | |
| cmake-init-cache: ${{ env.ONEFLOW_SRC }}/${{ matrix.cmake-init-cache }} | |
| 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 | |
| manylinux-cache-dir: ${{ env.MANYLINUX_CACHE_DIR }} | |
| docker-run-use-system-http-proxy: false | |
| docker-run-use-lld: true | |
| retry-failed-build: true | |
| clean-ccache: true | |
| compute-platform: ${{ env.COMPUTE_PLATFORM }} | |
| python-versions: | | |
| 3.8 | |
| 3.10 | |
| - name: Upload wheelhouse | |
| uses: ./.github/actions/upload_oss | |
| with: | |
| src_path: ${{ env.WHEELHOUSE_DIR }} | |
| oss_dst_path: oss://${{ env.OSS_BUCKET }}/${{ env.OSS_WHEEL_HOUSE_DIR }}/${{ env.COMPUTE_PLATFORM }} | |
| 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 -b ${{ env.OSS_BUCKET }} \ | |
| --dir_key ${{ env.OSS_WHEEL_HOUSE_DIR }}/${{ env.COMPUTE_PLATFORM }} \ | |
| --index_key=${{ env.OSS_WHEEL_HOUSE_DIR }}/${{ env.COMPUTE_PLATFORM }}/index.html \ | |
| --index_key=${{ env.OSS_GITHUB_REF_DIR}}/${{ env.COMPUTE_PLATFORM }}/index.html |