[Feat] Add torch.compile support #683
Workflow file for this run
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
| # https://docs.github.com/en/actions/publishing-packages/publishing-docker-images#publishing-images-to-github-packages | |
| # | |
| name: Docker image on ghcr.io | |
| on: | |
| push: | |
| tags: | |
| - 'v*' | |
| pull_request: | |
| branches: main | |
| schedule: | |
| - cron: '0 2 29 * *' # At 02:00 on day-of-month 29 | |
| env: | |
| REGISTRY: ghcr.io | |
| jobs: | |
| build-and-push-image: | |
| runs-on: ubuntu-latest | |
| strategy: | |
| fail-fast: false | |
| matrix: | |
| # Must match version at https://www.python.org/ftp/python/ | |
| python: ["3.10.13", "3.11.8", "3.12.7"] | |
| framework: ["tf", "torch", "tf,viz,html,contrib", "torch,viz,html,contrib"] | |
| # Sets the permissions granted to the `GITHUB_TOKEN` for the actions in this job. | |
| permissions: | |
| contents: read | |
| packages: write | |
| steps: | |
| - name: Checkout repository | |
| uses: actions/checkout@v4 | |
| - name: Log in to the Container registry | |
| uses: docker/login-action@v3 | |
| with: | |
| registry: ${{ env.REGISTRY }} | |
| username: ${{ github.actor }} | |
| password: ${{ secrets.GITHUB_TOKEN }} | |
| - name: Sanitize docker tag | |
| run: | |
| PREFIX_DOCKER_TAG="${{ matrix.framework }}-py${{ matrix.python }}-" | |
| PREFIX_DOCKER_TAG=$(echo PREFIX_DOCKER_TAG|sed 's/,/-/g') | |
| echo PREFIX_DOCKER_TAG=${PREFIX_DOCKER_TAG} >> $GITHUB_ENV | |
| - name: Extract metadata (tags, labels) for Docker | |
| id: meta | |
| uses: docker/metadata-action@v5 | |
| with: | |
| images: ${{ env.REGISTRY }}/${{ github.repository }} | |
| tags: | | |
| # used only on schedule event | |
| type=schedule,pattern={{date 'YYYY-MM'}},prefix=${{ env.PREFIX_DOCKER_TAG }} | |
| # used only if a tag following semver is published | |
| type=semver,pattern={{raw}},prefix=${{ env.PREFIX_DOCKER_TAG }} | |
| - name: Build Docker image | |
| id: build | |
| uses: docker/build-push-action@v6 | |
| with: | |
| context: . | |
| build-args: | | |
| FRAMEWORK=${{ matrix.framework }} | |
| PYTHON_VERSION=${{ matrix.python }} | |
| DOCTR_REPO=${{ github.repository }} | |
| DOCTR_VERSION=${{ github.sha }} | |
| push: false # push only if `import doctr` works | |
| tags: ${{ steps.meta.outputs.tags }} | |
| - name: Check if `import doctr` works | |
| run: docker run ${{ steps.build.outputs.imageid }} python3 -c 'import doctr' | |
| - name: Push Docker image | |
| # Push only if the CI is not triggered by "PR on main" | |
| if: ${{ (github.ref == 'refs/heads/main' && github.event_name != 'pull_request') || (startsWith(github.ref, 'refs/tags') && github.event_name == 'push') }} | |
| uses: docker/build-push-action@v6 | |
| with: | |
| context: . | |
| build-args: | | |
| FRAMEWORK=${{ matrix.framework }} | |
| PYTHON_VERSION=${{ matrix.python }} | |
| DOCTR_REPO=${{ github.repository }} | |
| DOCTR_VERSION=${{ github.sha }} | |
| push: true | |
| tags: ${{ steps.meta.outputs.tags }} |