TensorRT-RTX/.github/workflows/ci.yaml at 8f8103f24fc8ee56c80624dab900d3d3bb684774 · tburt-nv/TensorRT-RTX · 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
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
name: CI Build and Test
on: push
env:
  TRT_RTX_FILENAME: TensorRT-RTX-1.1.1.26.Linux.x86_64-gnu.cuda-12.9.tar.gz
  TRTRTX_INSTALL_DIR: /opt/tensorrt_rtx
  BUILD_DIR: build
  UV_PROJECT: .github/workflows/
jobs:
  build:
    runs-on: ubuntu-24.04
    container: nvcr.io/nvidia/cuda:12.9.1-devel-ubuntu22.04
    steps:
      - uses: actions/checkout@v5

      - name: Cache Debian packages
        id: cache-deb
        uses: actions/cache@v4
        with:
          path: /var/cache/apt/archives/
          key: ${{ runner.os }}-apt-build

      - name: Install uv
        uses: astral-sh/setup-uv@v6
        with:
          python-version: 3.9

      - name: Set up Python
        run: uv python install

      - name: Cache TensorRT RTX
        id: cache-trt-rtx
        uses: actions/cache@v4
        with:
          path: ${{ env.TRTRTX_INSTALL_DIR }}
          key: tensorrt-rtx-${{ runner.os }}-${{ env.TRT_RTX_FILENAME }}

      - name: Run build script
        env:
          CACHE_DEB_HIT: ${{ steps.cache-deb.outputs.cache-hit }}
          CACHE_TRT_RTX_HIT: ${{ steps.cache-trt-rtx.outputs.cache-hit }}
        run: uv run --group build .github/workflows/build.py

      - name: Upload build artifacts
        uses: actions/upload-artifact@v4
        with:
          name: build-artifacts
          path: |
            {{ $env.BUILD_DIR }}/

  test:
    # Disable while testing
    if: false
    needs: build
    runs-on: [linux-amd64-gpu-rtx2080-latest-1]
    container:
      image: nvcr.io/nvidia/cuda:12.9.1-runtime-ubuntu22.04
      options: --gpus all
    steps:
      - uses: actions/checkout@v5

      - name: Install uv
        uses: astral-sh/setup-uv@v6
        with:
          python-version: 3.9

      - name: Set up Python
        run: uv python install

      - name: Cache TensorRT RTX
        id: cache-trt-rtx
        uses: actions/cache@v4
        with:
          path: ${{ env.TRTRTX_INSTALL_DIR }}
          key: tensorrt-rtx-${{ runner.os }}-${{ env.TRT_RTX_FILENAME }}

      - name: Download build artifacts
        uses: actions/download-artifact@v4
        with:
          name: build-artifacts

      - name: Run test script
        env:
          CACHE_TRT_RTX_HIT: ${{ steps.cache-trt-rtx.outputs.cache-hit }}
        run: python3 .github/workflows/test.py