Module cubecl_core::prelude

Re-exports

• pub use crate::Kernel;
• pub use crate::RuntimeArg;
• pub use crate::codegen::KernelExpansion;
• pub use crate::codegen::KernelIntegrator;
• pub use crate::codegen::KernelSettings;
• pub use crate::compute::CompiledKernel;
• pub use crate::compute::CubeCount;
• pub use crate::compute::CubeTask;
• pub use crate::compute::KernelBuilder;
• pub use crate::compute::KernelLauncher;
• pub use crate::compute::KernelTask;
• pub use crate::frontend::cmma;
• pub use crate::ir::CubeDim;
• pub use crate::ir::KernelDefinition;
• pub use crate::frontend::Array;
• pub use crate::frontend::ArrayHandleRef;
• pub use crate::frontend::Bool;
• pub use crate::frontend::Float;
• pub use crate::frontend::LaunchArg;
• pub use crate::frontend::Slice;
• pub use crate::frontend::SliceMut;
• pub use crate::frontend::Tensor;
• pub use crate::frontend::TensorArg;
• pub use crate::frontend::UInt;
• pub use crate::frontend::F16;
• pub use crate::frontend::F32;
• pub use crate::frontend::F64;
• pub use crate::frontend::I32;
• pub use crate::frontend::I64;
• pub use crate::frontend::ABSOLUTE_POS;
• pub use crate::frontend::ABSOLUTE_POS;
• pub use crate::frontend::ABSOLUTE_POS_X;
• pub use crate::frontend::ABSOLUTE_POS_X;
• pub use crate::frontend::ABSOLUTE_POS_Y;
• pub use crate::frontend::ABSOLUTE_POS_Y;
• pub use crate::frontend::ABSOLUTE_POS_Z;
• pub use crate::frontend::ABSOLUTE_POS_Z;
• pub use crate::frontend::CUBE_COUNT;
• pub use crate::frontend::CUBE_COUNT;
• pub use crate::frontend::CUBE_COUNT_X;
• pub use crate::frontend::CUBE_COUNT_X;
• pub use crate::frontend::CUBE_COUNT_Y;
• pub use crate::frontend::CUBE_COUNT_Y;
• pub use crate::frontend::CUBE_COUNT_Z;
• pub use crate::frontend::CUBE_COUNT_Z;
• pub use crate::frontend::CUBE_DIM;
• pub use crate::frontend::CUBE_DIM;
• pub use crate::frontend::CUBE_DIM_X;
• pub use crate::frontend::CUBE_DIM_X;
• pub use crate::frontend::CUBE_DIM_Y;
• pub use crate::frontend::CUBE_DIM_Y;
• pub use crate::frontend::CUBE_DIM_Z;
• pub use crate::frontend::CUBE_DIM_Z;
• pub use crate::frontend::CUBE_POS;
• pub use crate::frontend::CUBE_POS;
• pub use crate::frontend::CUBE_POS_X;
• pub use crate::frontend::CUBE_POS_X;
• pub use crate::frontend::CUBE_POS_Y;
• pub use crate::frontend::CUBE_POS_Y;
• pub use crate::frontend::CUBE_POS_Z;
• pub use crate::frontend::CUBE_POS_Z;
• pub use crate::frontend::SUBCUBE_DIM;
• pub use crate::frontend::SUBCUBE_DIM;
• pub use crate::frontend::UNIT_POS;
• pub use crate::frontend::UNIT_POS;
• pub use crate::frontend::UNIT_POS_X;
• pub use crate::frontend::UNIT_POS_X;
• pub use crate::frontend::UNIT_POS_Y;
• pub use crate::frontend::UNIT_POS_Y;
• pub use crate::frontend::UNIT_POS_Z;
• pub use crate::frontend::UNIT_POS_Z;
• pub use crate::frontend::subcube_all;
• pub use crate::frontend::subcube_all;
• pub use crate::frontend::subcube_max;
• pub use crate::frontend::subcube_max;
• pub use crate::frontend::subcube_min;
• pub use crate::frontend::subcube_min;
• pub use crate::frontend::subcube_prod;
• pub use crate::frontend::subcube_prod;
• pub use crate::frontend::subcube_sum;
• pub use crate::frontend::subcube_sum;
• pub use crate::frontend::branch::*;
• pub use crate::frontend::synchronization::*;
• pub use crate::frontend::*;

Structs

• ComputeClient
  The ComputeClient is the entry point to require tasks from the ComputeServer. It should be obtained for a specific device via the Compute struct.

Traits

• CubeElement
  The base element trait for the jit backend.
• Runtime
  Runtime for the CubeCL.

Attribute Macros

• cube
  Derive macro for the module.

Derive Macros

• CubeLaunch
• CubeType