Add flops benchmark (#169) #150
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: Publish Docker image | |
| on: | |
| # Allow manual runs | |
| workflow_dispatch: | |
| # Only run for push on the main branch or for tagged version | |
| push: | |
| branches: | |
| - master | |
| tags: | |
| - v*.*.* | |
| env: | |
| REGISTRY: ghcr.io | |
| IMAGE_NAME: ${{ github.repository }} | |
| permissions: | |
| packages: write | |
| # define build arguments | |
| jobs: | |
| build-image: | |
| strategy: | |
| fail-fast: false | |
| matrix: | |
| include: | |
| - arch: cuda | |
| - arch: rocm | |
| runs-on: [self-hosted, "${{ matrix.arch }}"] | |
| permissions: | |
| contents: read | |
| packages: write | |
| steps: | |
| - name: Show all images | |
| run: | | |
| docker image ls | |
| - name: Prune | |
| run: | | |
| # Prune all images older than 2 weeks | |
| # The images are still on github registry | |
| docker image prune -f -a --filter "until=336h" | |
| docker system prune -f | |
| - name: Check out the repo | |
| uses: actions/checkout@v3 | |
| - name: Get Image Tag Name | |
| run: | | |
| REGEX="(.*)v(.*)\.(.*)\.(.*)" | |
| IMAGE_TAG="nightly" | |
| if [[ "${{ github.ref_name }}" =~ $REGEX ]]; then | |
| IMAGE_TAG="${GITHUB_REF_NAME##*/}" | |
| fi | |
| echo "IMAGE_TAG=$IMAGE_TAG" >> $GITHUB_ENV | |
| - name: Log in to the registry | |
| uses: docker/login-action@v2 | |
| with: | |
| registry: ${{ env.REGISTRY }} | |
| username: ${{ github.actor }} | |
| password: ${{ secrets.GITHUB_TOKEN }} | |
| - name: Extract metadata (tags, labels) for the image | |
| id: meta | |
| uses: docker/metadata-action@v4 | |
| with: | |
| images: ${{ env.REGISTRY }}/${{ env.IMAGE_NAME }} | |
| tags: | | |
| type=raw,value=${{ matrix.arch }}-${{ env.IMAGE_TAG }} | |
| - name: Build and push the image | |
| uses: docker/build-push-action@v3 | |
| with: | |
| context: . | |
| push: true | |
| file: docker/Dockerfile-${{ matrix.arch }} | |
| tags: ${{ steps.meta.outputs.tags }} | |
| labels: ${{ steps.meta.outputs.labels }} | |
| build-args: | | |
| ARCH=${{ matrix.arch }} | |
| CONFIG=standard.yaml |