Crate burn_cubecl

Crate burn_cubecl 

Source
Expand description

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.

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.