Struct cubecl_core::codegen::Execution

pub struct Execution<'h, K, R: Runtime, Scalars> { /* private fields */ }

Implementations

impl<'h, K, R: Runtime> Execution<'h, K, R, ()>

pub fn start( kernel: K, client: ComputeClient<R::Server, R::Channel>, ) -> Execution<'h, K, R, ()>

pub fn inputs( self, inputs: &'h [TensorHandleRef<'h, R>], ) -> Execution<'h, K, R, ()>

pub fn outputs( self, outputs: &'h [TensorHandleRef<'h, R>], ) -> Execution<'h, K, R, ()>

impl<'h, K, R> Execution<'h, K, R, ()>

where K: Kernel + 'static, R: Runtime,

pub fn with_scalars<E>(self, scalars: &[E]) -> Execution<'h, K, R, (&[E],)>

pub fn execute(self, launch: CubeCountSettings<R::Server>)

Execute a dynamic kernel.

impl<'h, 'a, K, R, E> Execution<'h, K, R, (&'a [E],)>

where K: Kernel + 'static, R: Runtime, E: CubeElement,

pub fn with_scalars<'b, E2>( self, scalars: &'b [E2], ) -> Execution<'h, K, R, (&'a [E], &'b [E2])>

pub fn execute(self, launch: CubeCountSettings<R::Server>)

Execute a dynamic kernel.

impl<'h, 'a, 'b, K, R, E1, E2> Execution<'h, K, R, (&'a [E1], &'b [E2])>

where K: Kernel + 'static, R: Runtime, E1: CubeElement, E2: CubeElement,

pub fn with_scalars<'c, E3>( self, scalars: &'c [E3], ) -> Execution<'h, K, R, (&'a [E1], &'b [E2], &'c [E3])>

pub fn execute(self, launch: CubeCountSettings<R::Server>)

where K: Kernel + 'static, R: Runtime,

Execute a dynamic kernel.

impl<'h, 'a, 'b, 'c, K, R, E1, E2, E3> Execution<'h, K, R, (&'a [E1], &'b [E2], &'c [E3])>

where K: Kernel + 'static, R: Runtime, E1: CubeElement, E2: CubeElement, E3: CubeElement,

pub fn execute(self, launch: CubeCountSettings<R::Server>)

Execute a dynamic kernel.