mirror of
https://github.com/huggingface/candle.git
synced 2025-06-15 10:26:33 +00:00
14010a8498
* Update our cuda runner. * Fix install rust. * Simplify. * Docker in docker. * Install curl * Install curl * No sudo. * devel * Put curl again. * Add missing deps. * pkg-config. * Cleanup.
34 lines
1.1 KiB
YAML
34 lines
1.1 KiB
YAML
name: CI / cuda
|
|
|
|
on:
|
|
workflow_dispatch:
|
|
pull_request:
|
|
|
|
jobs:
|
|
test-cuda:
|
|
concurrency:
|
|
group: ${{ github.workflow }}-${{ github.job }}-${{ github.head_ref || github.run_id }}
|
|
cancel-in-progress: true
|
|
runs-on: [single-gpu, nvidia-gpu, t4, ci]
|
|
container:
|
|
image: nvidia/cuda:12.3.1-devel-ubuntu22.04
|
|
options: --gpus 0
|
|
if: ${{ github.event.pull_request.head.repo.full_name == github.event.pull_request.base.repo.full_name }}
|
|
permissions:
|
|
contents: write
|
|
packages: write
|
|
# This is used to complete the identity challenge
|
|
# with sigstore/fulcio when running outside of PRs.
|
|
id-token: write
|
|
security-events: write
|
|
steps:
|
|
- name: Checkout repository
|
|
uses: actions/checkout@v3
|
|
- name: Install dependencies
|
|
run: apt-get update && apt install curl build-essential libssl-dev protobuf-compiler pkg-config -y
|
|
- name: Install Rust Stable
|
|
uses: actions-rust-lang/setup-rust-toolchain@v1
|
|
- uses: Swatinem/rust-cache@v2
|
|
- name: Test (cuda)
|
|
run: cargo test --features cuda
|