test-gpu.yaml

name: Rust GPU Tests

on:
  push:

concurrency:
  group: "${{ github.workflow }} @ ${{ github.event.pull_request.head.label || github.head_ref || github.ref }}"
  cancel-in-progress: true

jobs:
  e2e:
    runs-on: gpu
    container:
      image: ghcr.io/worldcoin/gpu-iris-mpc-base:1.0.0
    steps:
      - name: Checkout code
        uses: actions/checkout@v4

      - name: Cache Rust build
        uses: actions/cache@v3
        id: cache-rust
        with:
          path: |
            ~/.cargo/registry
            ~/.cargo/git
            target
          key: rust-build-${{ runner.os }}-${{ hashFiles('**/Cargo.lock') }}
          restore-keys: |
            rust-build-${{ runner.os }}-

      - name: Install Rust nightly
        uses: dtolnay/rust-toolchain@master
        with:
          toolchain: nightly

      - name: E2E Tests
        run: |
          export LD_LIBRARY_PATH=/usr/local/cuda-12.2/lib64:$LD_LIBRARY_PATH
          export CUDA_HOME=/usr/local/cuda-12.2
          /usr/local/cuda-12.2/bin/nvcc --version
          cargo test --release e2e
        shell: bash
        env:
          NCCL_P2P_DIRECT_DISABLE: 1
          NCCL_NET: Socket