Struct MlxCudaKernelConfig 

pub struct MlxCudaKernelConfig(/* private fields */);

Owned MLX-C CUDA kernel configuration handle.

Implementations

impl MlxCudaKernelConfig

pub fn new() -> Result<Self, MlxCError>

Creates an empty CUDA kernel configuration.

pub fn add_output( &mut self, shape: &[i32], dtype: MlxDtype, ) -> Result<(), MlxCError>

Adds one typed output with a static shape.

pub fn set_grid(&mut self, grid: [i32; 3]) -> Result<(), MlxCError>

Sets the launch grid dimensions.

pub fn set_thread_group( &mut self, thread_group: [i32; 3], ) -> Result<(), MlxCError>

Sets the thread-group dimensions.

pub fn set_init_value(&mut self, initial_value: f32) -> Result<(), MlxCError>

Sets the scalar initialization value for output buffers.

pub fn set_verbose(&mut self, verbose: bool) -> Result<(), MlxCError>

Toggles MLX’s verbose logging for this kernel launch.

pub fn add_dtype_template_argument( &mut self, name: &str, dtype: MlxDtype, ) -> Result<(), MlxCError>

Adds a dtype template argument.

pub fn add_int_template_argument( &mut self, name: &str, argument: i32, ) -> Result<(), MlxCError>

Adds an integer template argument.

pub fn add_bool_template_argument( &mut self, name: &str, argument: bool, ) -> Result<(), MlxCError>

Adds a boolean template argument.

Trait Implementations

impl Debug for MlxCudaKernelConfig

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

Formats the value using the given formatter.

impl Drop for MlxCudaKernelConfig

fn drop(&mut self)

Executes the destructor for this type.

fn pin_drop(self: Pin<&mut Self>)

🔬This is a nightly-only experimental API. (pin_ergonomics)

Execute the destructor for this type, but different to Drop::drop, it requires self to be pinned.