Struct KernelLauncher 

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

Prepare a kernel for launch.

Fields

settings: KernelSettings

Implementations

impl<R> KernelLauncher<R>

where R: Runtime,

pub fn with_scope<T>(&mut self, fun: impl FnMut(&mut Scope) -> T) -> T

pub fn register_scalar<C>(&mut self, scalar: C)

where C: ScalarArgType,

Register a scalar to be launched.

pub fn register_scalar_raw(&mut self, bytes: &[u8], dtype: StorageType)

Register a scalar to be launched from raw data.

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

where K: CubeKernel,

Launch the kernel.

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

where K: CubeKernel,

Launch the kernel without check bounds.

Safety

The kernel must not:

• Contain any out of bounds reads or writes. Doing so is immediate UB.
• Contain any loops that never terminate. These may be optimized away entirely or cause other unpredictable behaviour.

impl<R> KernelLauncher<R>

where R: Runtime,

pub fn register_tensor(&mut self, tensor: TensorArg<R>, ty: Type)

Push a new input tensor to the state.

pub fn register_array(&mut self, array: ArrayArg<R>, ty: Type)

Push a new input array to the state.

pub fn register_tensor_map<K>(&mut self, map: TensorMapArg<R, K>, ty: Type)

where K: TensorMapKind,

Push a new tensor to the state.

impl<R> KernelLauncher<R>

where R: Runtime,

pub fn new(settings: KernelSettings) -> KernelLauncher<R>