build-and-push-images.yml

name: Build and push images

on:
  push:
    branches: ["main"]
  pull_request:
    branches: ["main"]

env:
  REGISTRY: ghcr.io
  # github.repository as <account>/<repo>
  IMAGE_NAME_PREFIX: ${{ github.repository }}

jobs:

  build:

    runs-on: ubuntu-latest

    steps:
    - name: Checkout repository
      uses: actions/checkout@v3

    - name: Log into registry ${{ env.REGISTRY }}
      if: github.event_name != 'pull_request'
      uses: docker/login-action@28218f9b04b4f3f62068d7b6ce6ca5b26e35336c
      with:
        registry: ${{ env.REGISTRY }}
        username: ${{ github.actor }}
        password: ${{ secrets.GITHUB_TOKEN }}

    - name: build base image
      run: docker-compose build base

    - name: push base image
      run: docker push ${{ env.REGISTRY }}/${{ env.IMAGE_NAME_PREFIX }}/base:latest

    - name: build cpu image
      run: docker-compose build cpu

    - name: push cpu image
      run: docker push ${{ env.REGISTRY }}/${{ env.IMAGE_NAME_PREFIX }}/cpu:latest

    - name: build cpu-tpls image
      run: docker-compose build cpu-tpls

    - name: push cpu-tpls image
      run: docker push ${{ env.REGISTRY }}/${{ env.IMAGE_NAME_PREFIX }}/cpu-tpls:latest

    - name: build developer-cpu image
      run: docker-compose build --build-arg base_image=cpu-tpls developer-cpu

    - name: push developer-cpu image
      run: docker push ${{ env.REGISTRY }}/${{ env.IMAGE_NAME_PREFIX }}/developer-cpu:latest

    - name: build cuda-sm70 image
      run: |
        docker-compose build --build-arg cuda_arch_sm=70 cuda
        docker image tag ${{ env.REGISTRY }}/${{ env.IMAGE_NAME_PREFIX }}/cuda:latest \
           ${{ env.REGISTRY }}/${{ env.IMAGE_NAME_PREFIX }}/cuda-sm70:latest

    - name: push cuda-sm70 image
      run: docker push ${{ env.REGISTRY }}/${{ env.IMAGE_NAME_PREFIX }}/cuda-sm70:latest

    - name: build cuda-tpls-sm70 image
      run: |
        docker-compose build --build-arg cuda_arch_sm=70 cuda-tpls
        docker image tag ${{ env.REGISTRY }}/${{ env.IMAGE_NAME_PREFIX }}/cuda-tpls:latest \
           ${{ env.REGISTRY }}/${{ env.IMAGE_NAME_PREFIX }}/cuda-tpls-sm70:latest

    - name: push cuda-tpls-sm70 image
      run: docker push ${{ env.REGISTRY }}/${{ env.IMAGE_NAME_PREFIX }}/cuda-tpls-sm70:latest

    - name: build developer-cuda-sm70 image
      run: docker-compose build --build-arg base_image=cuda-tpls-sm70 developer-cuda-sm70

    - name: push developer-cuda-sm70 image
      run: docker push ${{ env.REGISTRY }}/${{ env.IMAGE_NAME_PREFIX }}/developer-cuda-sm70:latest