Skip to content

Merge branch 'main' into webgl #1517

Merge branch 'main' into webgl

Merge branch 'main' into webgl #1517

Workflow file for this run

name: GPU
on:
push:
branches: [ '**' ]
pull_request:
branches: [ "main" ]
env:
CARGO_TERM_COLOR: always
jobs:
build-opencl:
runs-on: self-hosted
steps:
- uses: actions/checkout@v4
- uses: dtolnay/rust-toolchain@stable
# - uses: Swatinem/rust-cache@v2
- name: Build (OpenCL)
run: cargo build --verbose --no-default-features --features opencl
build-cuda:
runs-on: self-hosted
steps:
- uses: actions/checkout@v4
- uses: dtolnay/rust-toolchain@stable
# - uses: Swatinem/rust-cache@v2
- name: Build (CUDA)
run: cargo build --verbose --no-default-features --features cuda
build-vulkan:
runs-on: self-hosted
steps:
- uses: actions/checkout@v4
- uses: dtolnay/rust-toolchain@stable
# - uses: Swatinem/rust-cache@v2
- name: Build (Vulkan)
run: cargo build --verbose --no-default-features --features vulkan
build-all:
runs-on: self-hosted
steps:
- uses: actions/checkout@v4
- uses: dtolnay/rust-toolchain@stable
# - uses: Swatinem/rust-cache@v2
- name: Build (cuda,opencl)
run: cargo build --verbose --no-default-features --features cuda,opencl
- name: Build (all)
run: cargo build --verbose --no-default-features --features cuda,opencl,vulkan
test-opencl:
runs-on: self-hosted
steps:
- uses: actions/checkout@v4
- uses: dtolnay/rust-toolchain@stable
# - uses: Swatinem/rust-cache@v2
- name: Run tests (OpenCL)
run: cargo test --verbose --no-default-features --features opencl
- name: Run tests (OpenCL, lazy)
run: cargo test --verbose --no-default-features --features opencl,lazy
- name: Run tests (OpenCL, autograd)
run: cargo test --verbose --no-default-features --features opencl,autograd
- name: Run tests (OpenCL, graph)
run: cargo test --verbose --no-default-features --features opencl,graph
- name: Run tests (OpenCL, modules)
run: cargo test --verbose --no-default-features --features opencl,autograd,graph,cpu
- name: Run tests (OpenCL, fork)
run: CUSTOS_USE_UNIFIED=false cargo test --verbose --no-default-features --features opencl,fork
test-cuda:
runs-on: self-hosted
steps:
- uses: actions/checkout@v4
- uses: dtolnay/rust-toolchain@stable
# - uses: Swatinem/rust-cache@v2
- name: Run tests (CUDA)
run: cargo test --verbose --no-default-features --features cuda
- name: Run tests (CUDA, lazy)
run: cargo test --verbose --no-default-features --features cuda,lazy
- name: Run tests (CUDA, autograd)
run: cargo test --verbose --no-default-features --features cuda,autograd
- name: Run tests (CUDA, graph)
run: cargo test --verbose --no-default-features --features cuda,graph
- name: Run tests (CUDA, cpu, serde)
run: cargo test --verbose --no-default-features --features cuda,cpu,serde
- name: Run tests (CUDA, modules)
run: cargo test --verbose --no-default-features --features cuda,autograd,graph,cpu
test-vulkan:
runs-on: self-hosted
steps:
- uses: actions/checkout@v4
- uses: dtolnay/rust-toolchain@stable
# - uses: Swatinem/rust-cache@v2
- name: Run tests (vulkan)
run: cargo test --verbose --no-default-features --features vulkan
- name: Run tests (vulkan, lazy)
run: cargo test --verbose --no-default-features --features vulkan,lazy
- name: Run tests (vulkan, autograd)
run: cargo test --verbose --no-default-features --features vulkan,autograd
- name: Run tests (vulkan, graph)
run: cargo test --verbose --no-default-features --features vulkan,graph
- name: Run tests (vulkan, cpu)
run: cargo test --verbose --no-default-features --features vulkan,cpu
- name: Run tests (vulkan, modules)
run: cargo test --verbose --no-default-features --features vulkan,autograd,graph,cpu
test-all:
runs-on: self-hosted
steps:
- uses: actions/checkout@v4
- uses: dtolnay/rust-toolchain@stable
# - uses: Swatinem/rust-cache@v2
- name: Run tests (CUDA, OpenCL, vulkan)
run: cargo test --verbose --no-default-features --features cuda,opencl,vulkan
- name: Run tests (CUDA, OpenCL, lazy, vulkan)
run: cargo test --verbose --no-default-features --features cuda,opencl,lazy,vulkan
- name: Run tests (CUDA, OpenCL, autograd, vulkan)
run: cargo test --verbose --no-default-features --features cuda,opencl,autograd,vulkan
- name: Run tests (CUDA, OpenCL, graph, vulkan)
run: cargo test --verbose --no-default-features --features cuda,opencl,graph,vulkan
- name: Run cuda (untyped device) tests
run: cargo test --verbose --no-default-features --features untyped,cuda,half