Struct cubecl_core::compute::KernelLauncher

pub struct KernelLauncher<R: Runtime> {
    pub settings: KernelSettings,
    /* private fields */
}

Prepare a kernel for launch.

Fields

settings: KernelSettings

Implementations

impl<R: Runtime> KernelLauncher<R>

pub fn register_tensor(&mut self, tensor: &TensorHandleRef<'_, R>)

Register a tensor to be launched.

pub fn register_array(&mut self, array: &ArrayHandleRef<'_, R>)

Register an array to be launched.

pub fn register_u32(&mut self, scalar: u32)

Register a u32 scalar to be launched.

pub fn register_i32(&mut self, scalar: i32)

Register a i32 scalar to be launched.

pub fn register_i64(&mut self, scalar: i64)

Register a i64 scalar to be launched.

pub fn register_bf16(&mut self, scalar: bf16)

Register a bf16 scalar to be launched.

pub fn register_f16(&mut self, scalar: f16)

Register a f16 scalar to be launched.

pub fn register_f32(&mut self, scalar: f32)

Register a f32 scalar to be launched.

pub fn register_f64(&mut self, scalar: f64)

Register a f64 scalar to be launched.

pub fn launch<K: Kernel>( self, cube_count: CubeCount<R::Server>, kernel: K, client: &ComputeClient<R::Server, R::Channel>, )

Launch the kernel.

pub unsafe fn launch_unchecked<K: Kernel>( self, cube_count: CubeCount<R::Server>, kernel: K, client: &ComputeClient<R::Server, R::Channel>, )

Launch the kernel without check bounds.

Safety

Out-of-bounds reads and writes can happen.

Trait Implementations

impl<R: Runtime> Default for KernelLauncher<R>

fn default() -> Self

Returns the “default value” for a type.