Struct KernelLaunchConfig 

pub struct KernelLaunchConfig {
    pub workgroup_size: (u32, u32, u32),
    pub workgroup_count: (u32, u32, u32),
    pub shared_memory_size: Option<usize>,
    pub stream_id: Option<usize>,
}

Kernel launch configuration

Fields

workgroup_size: (u32, u32, u32)

Workgroup size (local work size)

workgroup_count: (u32, u32, u32)

Number of workgroups (global work size / workgroup size)

shared_memory_size: Option<usize>

Shared memory size in bytes

stream_id: Option<usize>

Stream/queue for asynchronous execution

Implementations

impl KernelLaunchConfig

pub fn linear(global_size: u32, workgroup_size: Option<u32>) -> Self

Create a 1D launch configuration

pub fn grid_2d( global_size: (u32, u32), workgroup_size: Option<(u32, u32)>, ) -> Self

Create a 2D launch configuration

pub fn grid_3d( global_size: (u32, u32, u32), workgroup_size: Option<(u32, u32, u32)>, ) -> Self

Create a 3D launch configuration

pub fn with_shared_memory(self, size: usize) -> Self

Set shared memory size

pub fn with_stream(self, stream_id: usize) -> Self

Set execution stream

pub fn total_threads(&self) -> u64

Get total number of threads

Trait Implementations

impl Clone for KernelLaunchConfig

fn clone(&self) -> KernelLaunchConfig

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for KernelLaunchConfig

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

Formats the value using the given formatter.