Crate burn_wgpu

Modules

graphics

Structs

CubeBackend

Generic tensor backend that can be compiled just-in-time to any shader runtime

CubeDim

CubeTensor

The basic tensor primitive struct.

RuntimeOptions

The values that control how a WGPU Runtime will perform its calculations.

WgpuResource

The memory resource that can be allocated for wgpu.

WgpuRuntime

Runtime that uses the wgpu crate with the wgsl compiler. This is used in the Wgpu backend. For advanced configuration, use init_setup to pass in runtime options or to select a specific graphics API.

WgpuSetup

A complete setup used to run wgpu.

WgpuStorage

Buffer storage for wgpu.

flex32

A floating point type with relaxed precision, minimum f16, max [f32].

Enums

AutoCompiler

MemoryConfiguration

High level configuration of memory management.

WgpuDevice

The device struct when using the wgpu backend.

Traits

BoolElement

The element type for booleans for the jit backend.

FloatElement

The float element type for the jit backend.

IntElement

The int element type for the jit backend.

Functions

init_device

Create a WgpuDevice on an existing WgpuSetup. Useful when you want to share a device between CubeCL and other wgpu-dependent libraries.

init_setup

Like init_setup_async, but synchronous. On wasm, it is necessary to use init_setup_async instead.

init_setup_async

Initialize a client on the given device with the given options. This function is useful to configure the runtime options or to pick a different graphics API.

Type Aliases

Wgpufusion

Tensor backend that uses the wgpu crate for executing GPU compute shaders.