Crate burn_cubecl

Burn JIT Backend

Re-exports

pub use element::BoolElement;

pub use element::CubeElement;

pub use element::FloatElement;

pub use element::IntElement;

pub use cubecl;

Modules

element

Elements for JIT backend

fusionfusion

Module for interacting with fusion

kernel

Kernel module

ops

Utilities for implementing JIT kernels

tensor

Tensor module.

Macros

execute_with_dtype

Macro to execute a kernel/operation for a given element type.

Structs

CubeBackend

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

CubeTuneId

ID used to identify a Just-in-Time environment.

Enums

CubeAutotuneKey

Key for all autotune-enabled operations

Traits

CubeRuntime

Just-in-Time runtime extending the cube runtime.