Struct MlxCudaKernel 

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

Owned MLX-C CUDA kernel handle.

Implementations

impl MlxCudaKernel

pub fn new( kernel_name: &str, input_names: &[&str], output_names: &[&str], kernel_header: &str, kernel_source: &str, ensure_row_contiguous: bool, shared_memory_bytes: i32, ) -> Result<Self, MlxCError>

Compiles a CUDA kernel from source with named inputs and outputs.

Errors

Returns the captured MLX-C description when compilation fails.

pub fn apply( &self, context: &MlxBindingsContext, inputs: &MlxArrayVector, config: &MlxCudaKernelConfig, ) -> Result<MlxArrayVector, MlxCError>

Applies the kernel to the inputs on the given stream.

Errors

Returns the captured MLX-C description when the launch fails.

Trait Implementations

impl Debug for MlxCudaKernel

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

Formats the value using the given formatter.

impl Drop for MlxCudaKernel

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.