ubuntu.yml

name: Build Docker images (Ubuntu)

permissions:
  packages: write      

on:
  pull_request:
    paths:
      - 'cann/Dockerfile.ubuntu'
      - 'llm/**'
      - 'pytorch/**'
    branches:
      - main
  push:
    paths:
      - 'cann/Dockerfile.ubuntu'
      - 'llm/**'
      - 'pytorch/**'
    branches:
      - main

jobs:
  build-cann:
    runs-on: self-hosted
    steps:
      - name: Checkout
        uses: actions/checkout@v3

      - name: Build
        run: |
          docker build -t ascendai/cann:ubuntu2004 -f ./cann/Dockerfile.ubuntu .
          docker build -t ascendai/pytorch:ubuntu2004 -f ./pytorch/Dockerfile .
          cd llm
          docker build -t ascendai/llm:ubuntu2004 -f ./Dockerfile .
 
      - name: Push
        if: github.event_name == 'push'
        run: |
          docker push ascendai/cann:ubuntu2004
          docker push ascendai/pytorch:ubuntu2004
          docker push ascendai/llm:ubuntu2004