Struct WebGpuBackend 

pub struct WebGpuBackend { /* private fields */ }

Cross-platform GPU compute backend backed by wgpu.

Lifecycle

1. WebGpuBackend::new() — create an uninitialised backend.
2. init() — select the best available adapter and create the device.
3. Use alloc, copy_htod, compute ops, copy_dtoh, free.
4. synchronize() — wait for all pending GPU work to finish.

Implementations

impl WebGpuBackend

pub fn new() -> Self

Create a new, uninitialised WebGPU backend.

Trait Implementations

impl ComputeBackend for WebGpuBackend

fn name(&self) -> &str

Backend name (e.g., "cuda", "rocm", "metal").

fn init(&mut self) -> BackendResult<()>

Initialize the backend (select device, create context).

fn is_initialized(&self) -> bool

Returns true if the backend is ready for operations.

fn gemm( &self, _trans_a: BackendTranspose, _trans_b: BackendTranspose, m: usize, n: usize, k: usize, _alpha: f64, _a_ptr: u64, _lda: usize, _b_ptr: u64, _ldb: usize, _beta: f64, _c_ptr: u64, _ldc: usize, ) -> BackendResult<()>

General matrix multiply: C = alpha * op(A) * op(B) + beta * C.

fn conv2d_forward( &self, _input_ptr: u64, input_shape: &[usize], _filter_ptr: u64, filter_shape: &[usize], _output_ptr: u64, output_shape: &[usize], stride: &[usize], padding: &[usize], ) -> BackendResult<()>

2D convolution forward pass.

fn attention( &self, _q_ptr: u64, _k_ptr: u64, _v_ptr: u64, _o_ptr: u64, _batch: usize, _heads: usize, seq_q: usize, seq_kv: usize, head_dim: usize, scale: f64, _causal: bool, ) -> BackendResult<()>

Scaled dot-product attention.

fn reduce( &self, _op: ReduceOp, _input_ptr: u64, _output_ptr: u64, shape: &[usize], axis: usize, ) -> BackendResult<()>

Reduction along an axis.

fn unary( &self, _op: UnaryOp, _input_ptr: u64, _output_ptr: u64, n: usize, ) -> BackendResult<()>

Element-wise unary operation.

fn binary( &self, _op: BinaryOp, _a_ptr: u64, _b_ptr: u64, _output_ptr: u64, n: usize, ) -> BackendResult<()>

Element-wise binary operation.

fn synchronize(&self) -> BackendResult<()>

Synchronize all pending operations on this backend.

fn alloc(&self, bytes: usize) -> BackendResult<u64>

Allocate device memory.

fn free(&self, ptr: u64) -> BackendResult<()>

Free device memory previously allocated with alloc.

fn copy_htod(&self, dst: u64, src: &[u8]) -> BackendResult<()>

Copy data from host memory to device memory.

fn copy_dtoh(&self, dst: &mut [u8], src: u64) -> BackendResult<()>

Copy data from device memory to host memory.

impl Debug for WebGpuBackend

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl Default for WebGpuBackend

fn default() -> Self

Returns the “default value” for a type.