add CI

Signed-off-by: Tyler Burt <195370667+tburt-nv@users.noreply.github.com>

Author: tburt-nv

.github/workflows/build.py

@@ -0,0 +1,33 @@
+#!/usr/bin/env python3
+
+import os
+from utils import run_command, setup_trt_rtx, TRTRTX_INSTALL_DIR, BUILD_DIR
+
+def install_dependencies():
+    """Install required system packages."""
+    print("Installing dependencies...")
+    # Only install if cache miss
+    if os.environ.get('CACHE_DEB_HIT') != 'true':
+        run_command("apt-get update")
+        run_command("apt-get install -y cmake wget")
+
+def build_samples():
+    """Build C++ samples."""
+    print("Building C++ samples...")
+    run_command(f"cmake -B {BUILD_DIR} -S samples -DTRTRTX_INSTALL_DIR={TRTRTX_INSTALL_DIR}")
+    run_command(f"cmake --build {BUILD_DIR}")
+
+def main():
+    # Setup system
+    install_dependencies()
+
+    # Setup TensorRT RTX
+    setup_trt_rtx()
+
+    # Build samples
+    build_samples()
+
+    print("Build completed successfully!")
+
+if __name__ == "__main__":
+    main()

.github/workflows/ci.yaml

@@ -0,0 +1,76 @@
+name: CI Build and Test
+on: push
+env:
+  TRT_RTX_BASE_URL: https://developer.nvidia.com/downloads/trt/rtx_sdk/secure/1.1
+  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
+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 lsb_release
+        run: apt-get update && apt-get install -y python3
+
+      - 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: python3 .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/ai-workbench/python-cuda129:1.0.1
+      options: --gpus all
+    steps:
+      - uses: actions/checkout@v5
+
+      - name: Set up Python
+        uses: actions/setup-python@v6
+        with:
+          cache: 'pip'
+
+      - 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

.github/workflows/test.py

@@ -0,0 +1,52 @@
+#!/usr/bin/env python3
+
+import os
+from pathlib import Path
+from utils import run_command, setup_trt_rtx, TRTRTX_INSTALL_DIR, BUILD_DIR
+
+def install_python_deps():
+    """Install Python dependencies."""
+    print("Installing Python dependencies...")
+
+    # Install TensorRT RTX wheel
+    wheel_pattern = str(Path(TRTRTX_INSTALL_DIR) / "python" / "tensorrt_rtx-*-cp310-none-linux_x86_64.whl")
+    run_command(f"pip3 install {wheel_pattern}")
+
+    # Install sample requirements
+    run_command("pip3 install -r samples/helloWorld/python/requirements.txt")
+    run_command("pip3 install -r samples/apiUsage/python/requirements.txt")
+    run_command("pip3 install -r demo/flux1.dev/requirements.txt")
+    run_command("pip3 install -r demo/tests/requirements-test.txt")
+
+def run_cpp_tests():
+    """Run C++ sample tests."""
+    print("Running C++ tests...")
+    run_command(f"{BUILD_DIR}/helloWorld/cpp/helloWorld")
+    run_command(f"{BUILD_DIR}/apiUsage/cpp/apiUsage")
+
+def run_python_tests():
+    """Run Python sample tests."""
+    print("Running Python tests...")
+    # Set up environment for tests
+    test_env = os.environ.copy()
+    test_env["LD_LIBRARY_PATH"] = f"{TRTRTX_INSTALL_DIR}/lib:{test_env.get('LD_LIBRARY_PATH', '')}"
+
+    run_command("python3 samples/helloWorld/python/hello_world.py", env=test_env)
+    run_command("python3 samples/apiUsage/python/api_usage.py", env=test_env)
+    run_command("python3 -m pytest demo/tests -v", env=test_env)
+
+def main():
+    # Setup TensorRT RTX
+    setup_trt_rtx()
+
+    # Install Python dependencies
+    install_python_deps()
+
+    # Run tests
+    run_cpp_tests()
+    run_python_tests()
+
+    print("All tests completed successfully!")
+
+if __name__ == "__main__":
+    main()

.github/workflows/utils.py

@@ -0,0 +1,39 @@
+#!/usr/bin/env python3
+
+import os
+import subprocess
+import sys
+from pathlib import Path
+
+# Common environment variables
+TRT_RTX_BASE_URL = "https://developer.nvidia.com/downloads/trt/rtx_sdk/secure/1.1"
+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 = os.environ.get('BUILD_DIR', 'build')
+
+def run_command(cmd, check=True, shell=False, env=None):
+    """Run a command and handle errors."""
+    try:
+        subprocess.run(cmd if shell else cmd.split(), check=check, shell=shell, env=env)
+    except subprocess.CalledProcessError as e:
+        print(f"Error running command: {cmd}")
+        print(f"Exit code: {e.returncode}")
+        sys.exit(e.returncode)
+
+def setup_trt_rtx():
+    """Download and setup TensorRT RTX."""
+    if os.environ.get('CACHE_TRT_RTX_HIT') != 'true':
+        print("Cache miss for TensorRT RTX, downloading...")
+        url = f"{TRT_RTX_BASE_URL}/{TRT_RTX_FILENAME}"
+
+        if os.path.exists(TRTRTX_INSTALL_DIR):
+            print(f"Error: {TRTRTX_INSTALL_DIR} already exists. Remove it to proceed.")
+            exit(1)
+
+        # Download and extract
+        run_command(f"wget -O /tmp/trt.tar.gz {url}")
+        os.makedirs(TRTRTX_INSTALL_DIR)
+        run_command(f"tar --directory={TRTRTX_INSTALL_DIR} --strip-components=1 -xzf /tmp/trt.tar.gz")
+        os.remove("/tmp/trt.tar.gz")
+    else:
+        print("Cache hit for TensorRT RTX")