pub struct KernelLaunch<'a> { /* private fields */ }Expand description
2026-09-25: Builder for a kernel launch: grid, block, dynamic shared memory
and typed arguments. launch() passes the arguments as &[KernelArg] to
GpuBackend::launch_typed. Grid and block default to [1, 1, 1], shared
memory to 0.
Implementations§
Source§impl<'a> KernelLaunch<'a>
impl<'a> KernelLaunch<'a>
pub fn new(gpu: &'a dyn GpuBackend, kernel: KernelHandle) -> Self
pub fn grid(self, grid: [u32; 3]) -> Self
pub fn block(self, block: [u32; 3]) -> Self
Sourcepub fn arg_tensormap(self, map: &[u8; 128]) -> Self
pub fn arg_tensormap(self, map: &[u8; 128]) -> Self
2026-09-25: Add a 128-byte CUtensorMap by value, for a kernel parameter
declared __grid_constant__ const CUtensorMap. The bytes fill 16
consecutive slots and form one argument.
pub fn arg_u32(self, v: u32) -> Self
pub fn arg_u64(self, v: u64) -> Self
pub fn arg_i32(self, v: i32) -> Self
pub fn arg_f32(self, v: f32) -> Self
Auto Trait Implementations§
impl<'a> Freeze for KernelLaunch<'a>
impl<'a> !RefUnwindSafe for KernelLaunch<'a>
impl<'a> Send for KernelLaunch<'a>
impl<'a> Sync for KernelLaunch<'a>
impl<'a> Unpin for KernelLaunch<'a>
impl<'a> !UnwindSafe for KernelLaunch<'a>
Blanket Implementations§
Source§impl<T> BorrowMut<T> for Twhere
T: ?Sized,
impl<T> BorrowMut<T> for Twhere
T: ?Sized,
Source§fn borrow_mut(&mut self) -> &mut T
fn borrow_mut(&mut self) -> &mut T
Mutably borrows from an owned value. Read more