recursion.cuda.yml

name: CUDA Recursion Tests

on:
  push:
    branches: ["main"]
  pull_request:
    branches: ["**"]
    paths:
      - "crates/recursion/**"
      - ".github/workflows/recursion.cuda.yml"

concurrency:
  group: ${{ github.workflow }}-${{ github.event.pull_request.number || github.sha }}
  cancel-in-progress: true

env:
  CARGO_TERM_COLOR: always

jobs:
  cuda-backend:
    name: recursion cuda tracegen tests
    runs-on:
      - runs-on=${{ github.run_id }}/runner=test-gpu-nvidia/family=g6+g5+g6e
    steps:
      - uses: runs-on/action@v2
      - uses: actions/checkout@v6
        with:
          submodules: recursive
      - uses: dtolnay/rust-toolchain@stable
        env:
          RUSTUP_PERMIT_COPY_RENAME: "1"
      - uses: ./.github/actions/rust-cache-cuda
      - uses: taiki-e/install-action@nextest
      - name: Verify GPU setup
        run: |
          nvidia-smi
          nvidia-smi -L
          nvcc --version

      - name: Setup CUDA environment
        run: |
          # Find the actual CUDA library location
          CUDART_PATH=$(find /usr/local -name "libcudart.so" | head -1)
          if [ -n "$CUDART_PATH" ]; then
            CUDA_LIB_DIR=$(dirname "$CUDART_PATH")
            echo "CUDA libraries found at: $CUDA_LIB_DIR"
            echo "CUDA_LIB_DIR=$CUDA_LIB_DIR" >> $GITHUB_ENV
          else
            echo "CUDA libraries not found!"
            exit 1
          fi

      - name: Run recursion CUDA tests
        working-directory: crates/recursion
        run: cargo nextest run cuda --features cuda --test-threads=8