ubuntu-cuda.yml

name: Ubuntu CUDA

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

env:
  PYTHON_VERSION: '3.12'

jobs:
  build-ubuntu-cuda:
    strategy:
      matrix:
        include:
          # Ubuntu 24.04 + gcc-12 + CUDA 12.6.3
          - name: "Ubuntu 24.04 + gcc-12 + CUDA 12.6.3"
            os: ubuntu-24.04
            cuda: "12.6.3"
            compiler: gcc
            version: "12"

    runs-on: ${{ matrix.os }}
    name: 🐧 CUDA Build - ${{ matrix.name }}

    steps:
    - uses: actions/checkout@v4
      with:
        submodules: true
    - name: Setup Python
      uses: actions/setup-python@v5
      with:
        python-version: ${{ env.PYTHON_VERSION }}
    - name: Install packages
      run: sudo apt-get update && sudo apt-get install -yq python3-setuptools
    - name: Install CUDA
      env:
        cuda: ${{ matrix.cuda }}
      shell: bash
      run: |
        source ./.github/workflows/scripts/actions/install_cuda_ubuntu.sh
        echo "Adding CUDA to CUDA_PATH, PATH and LD_LIBRARY_PATH"
        echo "CUDA_PATH=${CUDA_PATH}" >> "$GITHUB_ENV"
        echo "${CUDA_PATH}/bin" >> "$GITHUB_PATH"
        echo "LD_LIBRARY_PATH=${CUDA_PATH}/lib64:${LD_LIBRARY_PATH}" >> "$GITHUB_ENV"
    - name: Configure Compiler
      run: |
        if [ "${{ matrix.compiler }}" = "gcc" ]; then
          echo "CC=gcc-${{ matrix.version }}" >> $GITHUB_ENV
          echo "CXX=g++-${{ matrix.version }}" >> $GITHUB_ENV
        else
          echo "CC=clang-${{ matrix.version }}" >> $GITHUB_ENV
          echo "CXX=clang++-${{ matrix.version }}" >> $GITHUB_ENV
        fi
    - name: Configure Build
      run: mkdir build && cd build && cmake -DUSE_CUDA=ON -DCMAKE_BUILD_TYPE=Release ..
    - name: Build
      run: cd build && make
    - name: Run CUDA Unit Test
      run: |
        if command -v nvidia-smi >/dev/null 2>&1 && nvidia-smi -L >/dev/null 2>&1; then
          /home/runner/work/CubbyFlow/CubbyFlow/build/bin/CUDATests
        else
          echo "No CUDA-capable GPU detected on runner. Skipping CUDA unit tests."
        fi