Struct rust_gpu_tools::cuda::Kernel

pub struct Kernel<'a> { /* private fields */ }

A kernel that can be executed.

Implementations

impl<'a> Kernel<'a>

pub fn arg<T: KernelArgument>(self, t: &'a T) -> Self

Set a kernel argument.

The arguments must live as long as the kernel. Hence make sure they are not dropped as long as the kernel is in use.

Example where this behaviour is enforced and leads to a compile-time error:

use rust_gpu_tools::cuda::Program;

fn would_break(program: &Program) {
   let data = vec![1, 2, 3, 4];
   let buffer = program.create_buffer_from_slice(&data).unwrap();
   let kernel = program.create_kernel("my_kernel", 4, 256).unwrap();
   let kernel = kernel.arg(&buffer);
   // This drop wouldn't error if the arguments wouldn't be bound to the kernels lifetime.
   drop(buffer);
   kernel.run().unwrap();
}

pub fn run(self) -> Result<(), GPUError>

Actually run the kernel.

Panics

Panics if the wrong number of arguments was provided.

Trait Implementations

impl Debug for Kernel<'_>

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.