docker.yml

name: Create and publish a Docker image

on:
  push:
    branches:
      - "master"
  release:
    types: [published]
  workflow_dispatch:
    inputs:
      tag:
        required: true
        type: string
        description: "Docker tag"
      latest:
        default: false
        type: boolean
        description: "Mark as latest"


env:
  REGISTRY: ghcr.io
  IMAGE_NAME: ${{ github.repository }}

jobs:
  build-and-push-image:
    runs-on: ubuntu-latest
    permissions:
      contents: read
      packages: write
    steps:
      - name: Checkout repository
        uses: actions/checkout@v3

      - name: Set up QEMU
        uses: docker/setup-qemu-action@v2

      - name: Set up Docker Buildx
        uses: docker/setup-buildx-action@v2

      - name: Log in to the Container registry
        uses: docker/login-action@f054a8b539a109f9f41c372932f1ae047eff08c9
        with:
          registry: ${{ env.REGISTRY }}
          username: ${{ github.actor }}
          password: ${{ secrets.GITHUB_TOKEN }}

      - name: Extract metadata (tags, labels) for Docker
        id: meta
        uses: docker/metadata-action@8e5442c4ef9f78752691e2d8f8d19755c6f78e81
        with:
          images: ${{ env.REGISTRY }}/${{ env.IMAGE_NAME }}

      - name: Tag for workflow_dispatch
        id: dispatch_tag
        run: |
          if [ x"$TAG" != x"" ];then
            echo "::set-output name=tag::${FULL_TAG}"
          fi
          if [ x"$LATEST" = x"true" ]; then
            echo "::set-output name=latest::${LATEST_TAG}"
          fi
        env:
          FULL_TAG: ${{ env.REGISTRY }}/${{ env.IMAGE_NAME }}:${{ github.event.inputs.tag }}
          LATEST_TAG: ${{ env.REGISTRY }}/${{ env.IMAGE_NAME }}:latest
          TAG: ${{ github.event.inputs.tag }}
          LATEST: ${{ github.event.inputs.latest }}

      - name: Build and push Docker image
        uses: docker/build-push-action@ad44023a93711e3deb337508980b4b5e9bcdc5dc
        with:
          context: .
          platforms: linux/amd64,linux/arm64
          push: true
          cache-from: type=gha
          cache-to: type=gha,mode=max
          tags: |
            ${{ steps.meta.outputs.tags }}
            ${{ steps.dispatch_tag.outputs.tag }}
            ${{ steps.dispatch_tag.outputs.latest }}
          labels: ${{ steps.meta.outputs.labels }}
  build-and-push-gpu-image:
    runs-on: ubuntu-latest
    permissions:
      contents: read
      packages: write
    steps:
      - name: Checkout repository
        uses: actions/checkout@v3

      # - name: Set up QEMU
      #   uses: docker/setup-qemu-action@v2

      - name: Set up Docker Buildx
        uses: docker/setup-buildx-action@v2

      - name: Log in to the Container registry
        uses: docker/login-action@f054a8b539a109f9f41c372932f1ae047eff08c9
        with:
          registry: ${{ env.REGISTRY }}
          username: ${{ github.actor }}
          password: ${{ secrets.GITHUB_TOKEN }}

      - name: Extract metadata (tags, labels) for Docker
        id: meta
        uses: docker/metadata-action@8e5442c4ef9f78752691e2d8f8d19755c6f78e81
        with:
          images: ${{ env.REGISTRY }}/${{ env.IMAGE_NAME }}
          flavor: |
            suffix=-cuda12,onlatest=true

      - name: Tag for workflow_dispatch
        id: dispatch_tag
        run: |
          if [ x"$TAG" != x"" ];then
            echo "::set-output name=tag::${FULL_TAG}"
          fi
          if [ x"$LATEST" = x"true" ]; then
            echo "::set-output name=latest::${LATEST_TAG}"
          fi
        env:
          FULL_TAG: ${{ env.REGISTRY }}/${{ env.IMAGE_NAME }}:${{ github.event.inputs.tag }}
          LATEST_TAG: ${{ env.REGISTRY }}/${{ env.IMAGE_NAME }}:latest
          TAG: ${{ github.event.inputs.tag }}
          LATEST: ${{ github.event.inputs.latest }}

      - name: Build and push Docker image
        uses: docker/build-push-action@ad44023a93711e3deb337508980b4b5e9bcdc5dc
        with:
          context: .
          platforms: linux/amd64,linux/arm64
          push: true
          no-cache: true
          build-args: |
            GPU=1
          tags: |
            ${{ steps.meta.outputs.tags }}
            ${{ steps.dispatch_tag.outputs.tag }}
            ${{ steps.dispatch_tag.outputs.latest }}
          labels: ${{ steps.meta.outputs.labels }}
  build-and-push-gpu-cuda13-image:
    runs-on: ubuntu-latest
    permissions:
      contents: read
      packages: write
    steps:
      - name: Checkout repository
        uses: actions/checkout@v3

      # - name: Set up QEMU
      #   uses: docker/setup-qemu-action@v2

      - name: Set up Docker Buildx
        uses: docker/setup-buildx-action@v2

      - name: Log in to the Container registry
        uses: docker/login-action@f054a8b539a109f9f41c372932f1ae047eff08c9
        with:
          registry: ${{ env.REGISTRY }}
          username: ${{ github.actor }}
          password: ${{ secrets.GITHUB_TOKEN }}

      - name: Extract metadata (tags, labels) for Docker
        id: meta
        uses: docker/metadata-action@8e5442c4ef9f78752691e2d8f8d19755c6f78e81
        with:
          images: ${{ env.REGISTRY }}/${{ env.IMAGE_NAME }}
          flavor: |
            suffix=-cuda13,onlatest=true

      - name: Tag for workflow_dispatch
        id: dispatch_tag
        run: |
          if [ x"$TAG" != x"" ];then
            echo "::set-output name=tag::${FULL_TAG}"
          fi
          if [ x"$LATEST" = x"true" ]; then
            echo "::set-output name=latest::${LATEST_TAG}"
          fi
        env:
          FULL_TAG: ${{ env.REGISTRY }}/${{ env.IMAGE_NAME }}:${{ github.event.inputs.tag }}-cuda13
          LATEST_TAG: ${{ env.REGISTRY }}/${{ env.IMAGE_NAME }}:latest-cuda13
          TAG: ${{ github.event.inputs.tag }}
          LATEST: ${{ github.event.inputs.latest }}

      - name: Build and push Docker image
        uses: docker/build-push-action@ad44023a93711e3deb337508980b4b5e9bcdc5dc
        with:
          context: .
          platforms: linux/amd64,linux/arm64
          push: true
          no-cache: true
          build-args: |
            GPU=1
            BUILDER_IMAGE_AMD64=ghcr.io/steineggerlab/build-containers:main-x86_64-manylinux_2_28-cuda13
            BUILDER_IMAGE_ARM64=ghcr.io/steineggerlab/build-containers:main-sbsa-cuda13
          tags: |
            ${{ steps.meta.outputs.tags }}
            ${{ steps.dispatch_tag.outputs.tag }}
            ${{ steps.dispatch_tag.outputs.latest }}
          labels: ${{ steps.meta.outputs.labels }}
  build-and-push-gpu-amd-image:
    runs-on: ubuntu-latest
    permissions:
      contents: read
      packages: write
    steps:
      - name: Checkout repository
        uses: actions/checkout@v3

      - name: Set up Docker Buildx
        uses: docker/setup-buildx-action@v2

      - name: Log in to the Container registry
        uses: docker/login-action@f054a8b539a109f9f41c372932f1ae047eff08c9
        with:
          registry: ${{ env.REGISTRY }}
          username: ${{ github.actor }}
          password: ${{ secrets.GITHUB_TOKEN }}

      - name: Extract metadata (tags, labels) for Docker
        id: meta
        uses: docker/metadata-action@8e5442c4ef9f78752691e2d8f8d19755c6f78e81
        with:
          images: ${{ env.REGISTRY }}/${{ env.IMAGE_NAME }}
          flavor: |
            suffix=-rocm7,onlatest=false

      - name: Tag for workflow_dispatch
        id: dispatch_tag
        run: |
          if [ x"$TAG" != x"" ];then
            echo "::set-output name=tag::${FULL_TAG}"
          fi
        env:
          FULL_TAG: ${{ env.REGISTRY }}/${{ env.IMAGE_NAME }}:${{ github.event.inputs.tag }}-rocm7-debian
          TAG: ${{ github.event.inputs.tag }}

      - name: Build and push Docker image
        uses: docker/build-push-action@ad44023a93711e3deb337508980b4b5e9bcdc5dc
        with:
          context: .
          file: .github/workflows/Dockerfile.GPU-debian-slim-rocm7
          platforms: linux/amd64
          push: true
          cache-from: type=gha
          cache-to: type=gha,mode=max
          tags: |
            ${{ steps.meta.outputs.tags }}
            ${{ steps.dispatch_tag.outputs.tag }}
          labels: ${{ steps.meta.outputs.labels }}