Struct WebGpuBackend 

pub struct WebGpuBackend { /* private fields */ }

Implementations

impl WebGpuBackend

pub fn new(device_ordinal: usize) -> Result<Self>

Initialize a WebGPU backend for a discrete GPU ordinal.

Examples

use tenferro_gpu::webgpu::WebGpuBackend;

let _ctor: fn(usize) -> tenferro_tensor::Result<WebGpuBackend> = WebGpuBackend::new;

Errors

Returns crate::Error::RuntimeState when no adapter/device is available, or crate::Error::BackendSource when CubeCL initialization fails.

pub fn new_default() -> Result<Self>

Initialize a WebGPU backend using CubeCL’s default adapter selection.

Examples

use tenferro_gpu::webgpu::WebGpuBackend;

let _ctor: fn() -> tenferro_tensor::Result<WebGpuBackend> = WebGpuBackend::new_default;

Errors

Returns crate::Error::RuntimeState when default adapter selection is unavailable, or crate::Error::BackendSource when initialization fails.

pub fn from_runtime(runtime: WebGpuRuntime) -> Self

Build a WebGPU backend from an initialized runtime.

Examples

use tenferro_gpu::{webgpu::WebGpuBackend, webgpu::WebGpuRuntime};

let _from_runtime: fn(WebGpuRuntime) -> WebGpuBackend = WebGpuBackend::from_runtime;

pub fn runtime(&self) -> &WebGpuRuntime

Return this backend’s WebGPU runtime.

Examples

use tenferro_gpu::{webgpu::WebGpuBackend, webgpu::WebGpuRuntime};

let _runtime: fn(&WebGpuBackend) -> &WebGpuRuntime = WebGpuBackend::runtime;

pub fn runtime_identity(&self) -> WebGpuRuntimeIdentity

Return the opaque identity of this exact executable backend instance.

Clones of a backend return the same identity. Independently initialized backends return different identities even when they target the same WebGPU device ordinal. This also covers Apple-backed WebGPU runtimes.

Examples

use tenferro_gpu::webgpu::WebGpuBackend;

let _identity = WebGpuBackend::runtime_identity;

pub fn synchronize(&self) -> Result<()>

Block until queued WebGPU work completes.

Examples

use tenferro_gpu::webgpu::WebGpuBackend;

let _sync: fn(&WebGpuBackend) -> tenferro_tensor::Result<()> = WebGpuBackend::synchronize;

Errors

Returns crate::Error::BackendSource when queue flush or synchronization fails, or crate::Error::RuntimeState when the runtime has lost its device state.

Trait Implementations

impl BackendCachedDot for WebGpuBackend

fn dot_general_read_into_accum_cached( &mut self, _cache: &mut Self::RuntimeCache, _cache_slot: Option<usize>, lhs: TensorRead<'_>, rhs: TensorRead<'_>, config: &DotGeneralConfig, accumulation: DotGeneralAccumulation, out: TensorWrite<'_>, ) -> Result<(), Error>

Apply cached scaled dot-general accumulation into caller-provided output.

impl BackendRuntimeCache for WebGpuBackend

impl BackendSession for WebGpuBackend

fn vdot_read( &mut self, _lhs: TensorRead<'_>, _rhs: TensorRead<'_>, ) -> Result<Tensor, Error>

Compute the all-axis conjugating dot product without transferring either input.

fn norm_squared_read(&mut self, _input: TensorRead<'_>) -> Result<Tensor, Error>

Compute the all-axis sum of squared magnitudes without taking a square root.

fn axpby_read_into_accum( &mut self, _alpha: ContractionScalar, _x: TensorRead<'_>, _beta: ContractionScalar, _y: TensorWrite<'_>, ) -> Result<(), Error>

Apply y <- alpha * x + beta * y in one pass into caller-owned storage.

impl BackendSessionHost for WebGpuBackend

fn with_backend_session<R: Send>( &mut self, f: impl FnOnce(&mut dyn BackendSession) -> R + Send, ) -> R

impl Clone for WebGpuBackend

fn clone(&self) -> WebGpuBackend

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for WebGpuBackend

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

Formats the value using the given formatter.

impl DotGeneralPreparation for WebGpuBackend

fn prepare( &self, request: DotGeneralPrepareRequest<'_>, ) -> Result<PrepareCapability, PrepareError>

Prepare one dot-general operation.

impl TensorAnalytic for WebGpuBackend

fn exp(&mut self, _input: &Tensor) -> Result<Tensor>

Errors

fn log(&mut self, _input: &Tensor) -> Result<Tensor>

Errors

fn sin(&mut self, _input: &Tensor) -> Result<Tensor>

Errors

fn cos(&mut self, _input: &Tensor) -> Result<Tensor>

Errors

fn tanh(&mut self, _input: &Tensor) -> Result<Tensor>

Errors

fn sqrt(&mut self, _input: &Tensor) -> Result<Tensor>

Errors

fn rsqrt(&mut self, _input: &Tensor) -> Result<Tensor>

Errors

fn pow(&mut self, _lhs: &Tensor, _rhs: &Tensor) -> Result<Tensor>

Errors

fn expm1(&mut self, _input: &Tensor) -> Result<Tensor>

Errors

fn log1p(&mut self, _input: &Tensor) -> Result<Tensor>

Errors

fn exp_read(&mut self, input: TensorRead<'_>) -> Result<Tensor, Error>

Errors

fn log_read(&mut self, input: TensorRead<'_>) -> Result<Tensor, Error>

Errors

fn sin_read(&mut self, input: TensorRead<'_>) -> Result<Tensor, Error>

Errors

fn cos_read(&mut self, input: TensorRead<'_>) -> Result<Tensor, Error>

Errors

fn tanh_read(&mut self, input: TensorRead<'_>) -> Result<Tensor, Error>

Errors

fn sqrt_read(&mut self, input: TensorRead<'_>) -> Result<Tensor, Error>

Errors

fn rsqrt_read(&mut self, input: TensorRead<'_>) -> Result<Tensor, Error>

Errors

fn pow_read( &mut self, lhs: TensorRead<'_>, rhs: TensorRead<'_>, ) -> Result<Tensor, Error>

Errors

fn expm1_read(&mut self, input: TensorRead<'_>) -> Result<Tensor, Error>

Errors

fn log1p_read(&mut self, input: TensorRead<'_>) -> Result<Tensor, Error>

Errors

impl TensorBackend for WebGpuBackend

impl TensorBuffer for WebGpuBackend

fn reclaim_buffer(&mut self, _tensor: Tensor)

impl TensorDeviceTransfer for WebGpuBackend

fn download_to_host(&mut self, tensor: TensorRead<'_>) -> Result<Tensor>

Explicitly copy a provider-owned read target into host storage.

fn upload_host_tensor(&mut self, tensor: TensorRead<'_>) -> Result<Tensor>

Explicitly copy a host read target into provider storage.

impl TensorDot for WebGpuBackend

fn dot_general( &mut self, lhs: &Tensor, rhs: &Tensor, config: &DotGeneralConfig, ) -> Result<Tensor>

Errors

fn dot_general_read_into( &mut self, lhs: TensorRead<'_>, rhs: TensorRead<'_>, config: &DotGeneralConfig, out: TensorWrite<'_>, ) -> Result<(), Error>

Overwrite caller-provided output with dot-general from read inputs.

fn dot_general_read_into_accum( &mut self, lhs: TensorRead<'_>, rhs: TensorRead<'_>, config: &DotGeneralConfig, accumulation: DotGeneralAccumulation, out: TensorWrite<'_>, ) -> Result<(), Error>

Apply scaled dot-general accumulation into caller-provided output.

impl TensorElementwise for WebGpuBackend

fn add(&mut self, _lhs: &Tensor, _rhs: &Tensor) -> Result<Tensor>

Errors

fn sub(&mut self, _lhs: &Tensor, _rhs: &Tensor) -> Result<Tensor>

Errors

fn mul(&mut self, _lhs: &Tensor, _rhs: &Tensor) -> Result<Tensor>

Errors

fn neg(&mut self, _input: &Tensor) -> Result<Tensor>

Errors

fn conj(&mut self, _input: &Tensor) -> Result<Tensor>

Errors

fn div(&mut self, _lhs: &Tensor, _rhs: &Tensor) -> Result<Tensor>

Errors

fn abs(&mut self, _input: &Tensor) -> Result<Tensor>

Errors

fn sign(&mut self, _input: &Tensor) -> Result<Tensor>

Errors

fn maximum(&mut self, _lhs: &Tensor, _rhs: &Tensor) -> Result<Tensor>

Errors

fn minimum(&mut self, _lhs: &Tensor, _rhs: &Tensor) -> Result<Tensor>

Errors

fn compare( &mut self, _lhs: &Tensor, _rhs: &Tensor, _dir: &CompareDir, ) -> Result<Tensor>

Errors

fn select( &mut self, _pred: &Tensor, _on_true: &Tensor, _on_false: &Tensor, ) -> Result<Tensor>

Errors

fn clamp( &mut self, _input: &Tensor, _lower: &Tensor, _upper: &Tensor, ) -> Result<Tensor>

Errors

fn elementwise_read_into( &mut self, op: ElementwiseReadOp, inputs: &[TensorRead<'_>], out: TensorWrite<'_>, ) -> Result<(), Error>

Execute an elementwise operation into caller-owned storage.

fn add_read( &mut self, lhs: TensorRead<'_>, rhs: TensorRead<'_>, ) -> Result<Tensor, Error>

Elementwise addition accepting either owned tensors or borrowed views.

fn add_into( &mut self, lhs: &Tensor, rhs: &Tensor, out: TensorWrite<'_>, ) -> Result<(), Error>

Overwrite caller-provided output with elementwise addition.

fn add_read_into( &mut self, lhs: TensorRead<'_>, rhs: TensorRead<'_>, out: TensorWrite<'_>, ) -> Result<(), Error>

Overwrite caller-provided output with elementwise addition from reads.

fn sub_read( &mut self, lhs: TensorRead<'_>, rhs: TensorRead<'_>, ) -> Result<Tensor, Error>

Elementwise subtraction accepting either owned tensors or borrowed views.

fn sub_into( &mut self, lhs: &Tensor, rhs: &Tensor, out: TensorWrite<'_>, ) -> Result<(), Error>

Overwrite caller-provided output with elementwise subtraction.

fn sub_read_into( &mut self, lhs: TensorRead<'_>, rhs: TensorRead<'_>, out: TensorWrite<'_>, ) -> Result<(), Error>

Overwrite caller-provided output with elementwise subtraction from reads.

fn mul_read( &mut self, lhs: TensorRead<'_>, rhs: TensorRead<'_>, ) -> Result<Tensor, Error>

Errors

fn mul_into( &mut self, lhs: &Tensor, rhs: &Tensor, out: TensorWrite<'_>, ) -> Result<(), Error>

Overwrite caller-provided output with elementwise multiplication.

fn mul_read_into( &mut self, lhs: TensorRead<'_>, rhs: TensorRead<'_>, out: TensorWrite<'_>, ) -> Result<(), Error>

Overwrite caller-provided output with elementwise multiplication from reads.

fn neg_read(&mut self, input: TensorRead<'_>) -> Result<Tensor, Error>

Errors

fn neg_into( &mut self, input: &Tensor, out: TensorWrite<'_>, ) -> Result<(), Error>

Overwrite caller-provided output with elementwise negation.

fn neg_read_into( &mut self, input: TensorRead<'_>, out: TensorWrite<'_>, ) -> Result<(), Error>

Overwrite caller-provided output with elementwise negation from a read.

fn conj_read(&mut self, input: TensorRead<'_>) -> Result<Tensor, Error>

Errors

fn conj_into( &mut self, input: &Tensor, out: TensorWrite<'_>, ) -> Result<(), Error>

Overwrite caller-provided output with elementwise conjugation.

fn conj_read_into( &mut self, input: TensorRead<'_>, out: TensorWrite<'_>, ) -> Result<(), Error>

Overwrite caller-provided output with elementwise conjugation from a read.

fn div_read( &mut self, lhs: TensorRead<'_>, rhs: TensorRead<'_>, ) -> Result<Tensor, Error>

Errors

fn div_into( &mut self, lhs: &Tensor, rhs: &Tensor, out: TensorWrite<'_>, ) -> Result<(), Error>

Overwrite caller-provided output with elementwise division.

fn div_read_into( &mut self, lhs: TensorRead<'_>, rhs: TensorRead<'_>, out: TensorWrite<'_>, ) -> Result<(), Error>

Overwrite caller-provided output with elementwise division from reads.

fn rem(&mut self, lhs: &Tensor, _rhs: &Tensor) -> Result<Tensor, Error>

Elementwise remainder.

fn rem_read( &mut self, lhs: TensorRead<'_>, rhs: TensorRead<'_>, ) -> Result<Tensor, Error>

Elementwise remainder accepting owned tensors or borrowed views.

fn abs_read(&mut self, input: TensorRead<'_>) -> Result<Tensor, Error>

Errors

fn sign_read(&mut self, input: TensorRead<'_>) -> Result<Tensor, Error>

Errors

fn maximum_read( &mut self, lhs: TensorRead<'_>, rhs: TensorRead<'_>, ) -> Result<Tensor, Error>

Errors

fn minimum_read( &mut self, lhs: TensorRead<'_>, rhs: TensorRead<'_>, ) -> Result<Tensor, Error>

Errors

fn compare_read( &mut self, lhs: TensorRead<'_>, rhs: TensorRead<'_>, dir: &CompareDir, ) -> Result<Tensor, Error>

Errors

fn select_read( &mut self, pred: TensorRead<'_>, on_true: TensorRead<'_>, on_false: TensorRead<'_>, ) -> Result<Tensor, Error>

Errors

fn clamp_read( &mut self, input: TensorRead<'_>, lower: TensorRead<'_>, upper: TensorRead<'_>, ) -> Result<Tensor, Error>

Errors

impl TensorFusion for WebGpuBackend

impl TensorIndexing for WebGpuBackend

fn gather( &mut self, _operand: &Tensor, _start_indices: &Tensor, _config: &GatherConfig, ) -> Result<Tensor>

Errors

fn scatter( &mut self, _operand: &Tensor, _scatter_indices: &Tensor, _updates: &Tensor, _config: &ScatterConfig, ) -> Result<Tensor>

Errors

fn slice(&mut self, _input: &Tensor, _config: &SliceConfig) -> Result<Tensor>

Errors

fn dynamic_slice( &mut self, _input: &Tensor, _starts: &Tensor, _slice_sizes: &[usize], ) -> Result<Tensor>

Errors

fn dynamic_update_slice( &mut self, _operand: &Tensor, _update: &Tensor, _starts: &Tensor, ) -> Result<Tensor>

Errors

fn pad(&mut self, _input: &Tensor, _config: &PadConfig) -> Result<Tensor>

Errors

fn concatenate(&mut self, _inputs: &[&Tensor], _axis: usize) -> Result<Tensor>

Errors

fn reverse(&mut self, _input: &Tensor, _axes: &[usize]) -> Result<Tensor>

Errors

impl TensorReduction for WebGpuBackend

fn reduce_sum(&mut self, _input: &Tensor, _axes: &[usize]) -> Result<Tensor>

Errors

fn reduce_prod(&mut self, _input: &Tensor, _axes: &[usize]) -> Result<Tensor>

Errors

fn reduce_max(&mut self, _input: &Tensor, _axes: &[usize]) -> Result<Tensor>

Errors

fn reduce_min(&mut self, _input: &Tensor, _axes: &[usize]) -> Result<Tensor>

Errors

fn reduce_sum_read( &mut self, input: TensorRead<'_>, axes: &[usize], ) -> Result<Tensor, Error>

Sum elements across axes from an owned tensor or borrowed view.

fn reduce_prod_read( &mut self, input: TensorRead<'_>, axes: &[usize], ) -> Result<Tensor, Error>

Multiply elements across axes from an owned tensor or borrowed view.

fn reduce_max_read( &mut self, input: TensorRead<'_>, axes: &[usize], ) -> Result<Tensor, Error>

Take maximum values across axes from an owned tensor or borrowed view.

fn reduce_min_read( &mut self, input: TensorRead<'_>, axes: &[usize], ) -> Result<Tensor, Error>

Take minimum values across axes from an owned tensor or borrowed view.

impl TensorStructural for WebGpuBackend

fn to_contiguous_read(&mut self, input: TensorRead<'_>) -> Result<Tensor>

Materialize an owned tensor or borrowed view into fresh compact storage.

fn copy_read_into( &mut self, _src: TensorRead<'_>, _dst: TensorWrite<'_>, ) -> Result<()>

Overwrite caller-provided storage from a readable tensor or view.

fn transpose(&mut self, input: &Tensor, perm: &[usize]) -> Result<Tensor>

Errors

fn reshape(&mut self, _input: &Tensor, _shape: &[usize]) -> Result<Tensor>

Errors

fn broadcast_in_dim( &mut self, _input: &Tensor, _shape: &[usize], _dims: &[usize], ) -> Result<Tensor>

Errors

fn cast(&mut self, _input: &Tensor, _to: DType) -> Result<Tensor>

Cast a tensor to another dtype using explicit dtype projection.

fn extract_diagonal( &mut self, _input: &Tensor, _axis_a: usize, _axis_b: usize, ) -> Result<Tensor>

Errors

fn embed_diagonal( &mut self, _input: &Tensor, _axis_a: usize, _axis_b: usize, ) -> Result<Tensor>

Errors

fn tril(&mut self, _input: &Tensor, _k: i64) -> Result<Tensor>

Errors

fn triu(&mut self, _input: &Tensor, _k: i64) -> Result<Tensor>

Errors

fn transpose_read( &mut self, input: TensorRead<'_>, perm: &[usize], ) -> Result<Tensor, Error>

Errors

fn reshape_read( &mut self, input: TensorRead<'_>, shape: &[usize], ) -> Result<Tensor, Error>

Errors

fn broadcast_in_dim_read( &mut self, input: TensorRead<'_>, shape: &[usize], dims: &[usize], ) -> Result<Tensor, Error>

Errors

fn convert(&mut self, input: &Tensor, to: DType) -> Result<Tensor, Error>

Convert a tensor to another dtype using checked dtype conversion.

impl TensorViewCanonicalization<f32, DynRank> for WebGpuBackend

fn to_contiguous( &mut self, view: &TypedTensorView<'_, f32>, ) -> Result<TypedTensor<f32>>

Errors

fn copy_into( &mut self, _src: &TypedTensorView<'_, f32>, _dst: &mut TypedTensorViewMut<'_, f32>, ) -> Result<()>

Errors