Skip to content

test

test #75

Workflow file for this run

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
steps:
- name: Checkout code
uses: actions/checkout@v4
- name: Validate presence of GPU devices
run: nvidia-smi
- name: Install OpenSSL && pkg-config
run: sudo apt-get update && sudo apt-get install -y pkg-config libssl-dev
- name: Install CUDA and NCCL dependencies
run: |
wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/cuda-keyring_1.1-1_all.deb
sudo dpkg -i cuda-keyring_1.1-1_all.deb
sudo apt-get update
sudo apt-get install -y cuda
sudo apt-get install -y git build-essential devscripts debhelper fakeroot
git clone https://github.com/NVIDIA/nccl.git
cd nccl
make -j24 pkg.debian.build
sudo dpkg -i build/pkg/deb/libnccl*.deb
- 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: cargo test --release e2e
shell: bash
env:
NCCL_P2P_DIRECT_DISABLE: 1
NCCL_NET: Socket