triton-dev-containers/.github/workflows/nvidia-image.yml at main · cmagina/triton-dev-containers · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
name: nvidia-image

on: # yamllint disable-line rule:truthy
  push:
    branches:
      - main
    paths:
      - .github/workflows/nvidia-image.yml
      - dockerfiles/Dockerfile.triton
      - entrypoint.sh
      - users.sh
  pull_request:
    paths:
      - .github/workflows/nvidia-image.yml
      - dockerfiles/Dockerfile.triton
      - entrypoint.sh
      - users.sh
  schedule:
    - cron: '0 0 * * 0' # Runs every Sunday at midnight UTC
  workflow_dispatch:

jobs:
  nvidia-image-build:
    uses: ./.github/workflows/image-build.yml
    with:
      image-name: nvidia
      image-tag: latest
      dockerfile: dockerfiles/Dockerfile.triton
      build-args: |
        USERNAME=triton
        CUSTOM_LLVM=false
        INSTALL_CUDNN=true
      platforms: linux/amd64
    secrets:
      qt_username: ${{ secrets.qt_username }}
      qt_password: ${{ secrets.qt_password }}

  nvidia-image-sbom:
    needs: nvidia-image-build
    if: ${{ github.event_name == 'push' || github.event_name == 'workflow_dispatch' }}
    uses: ./.github/workflows/image-sbom.yml
    with:
      image-name: nvidia
      image-tag: latest
    secrets:
      qt_username: ${{ secrets.qt_username }}
      qt_password: ${{ secrets.qt_password }}