rlx-wgpu 0.2.6

Cross-platform GPU backend for RLX via wgpu (Metal/Vulkan/DX12/WebGPU)
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62

// RLX — versatile ML compiler + runtime.
// Copyright (C) 2026 Eugene Hauptmann, Nataliya Kosmyna.
//
// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, version 3.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
// GNU General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with this program. If not, see <https://www.gnu.org/licenses/>.

//! RLX wgpu backend — cross-platform GPU execution via the `wgpu`
//! Rust crate (Metal on macOS, Vulkan on Linux, DX12 on Windows,
//! WebGPU in browsers).
//!
//! Compared to rlx-metal: same overall shape (device singleton, buffer
//! arena, per-op compute pipelines, command-buffer-per-forward-pass)
//! but with WGSL kernels and the wgpu Rust API instead of MSL + the
//! `metal` crate. Pure Rust deps — no FFI / submodules to manage.
//!
//! Layout:
//! - `device`   — wgpu instance/adapter/device singleton (sync wrapper)
//! - `buffer`   — typed GPU buffer + arena
//! - `kernels`  — WGSL source strings + per-kernel pipeline cache
//! - `backend`  — Backend trait impl + per-op dispatch

pub mod backend;
pub mod buffer;
pub mod calibrate;
pub mod coop_f16_vk;
pub mod device;
pub mod fft_dispatch;
pub mod fft_host;
pub mod gdn_host;
pub mod gguf_host;
pub mod im2col_host;
pub mod kernels;
pub mod llada2_gate_host;
pub mod log_mel_host;
#[cfg(feature = "splat")]
pub mod splat;
#[cfg(feature = "native-splat")]
pub mod splat_native;
pub mod training_bwd_host;
pub mod umap_knn_host;
pub mod unfuse;
pub mod welch_peaks_dispatch;
pub mod welch_peaks_host;

/// True if a wgpu adapter is reachable on this system. Always
/// available at the crate level; the runtime registry only registers
/// the backend when this returns `true` so tests on weird CI machines
/// without a GPU don't trip up.
pub fn is_available() -> bool {
    device::wgpu_device().is_some()
}

pub use device::{is_vulkan_available, select_vulkan_backend};