pub struct MlxMetalKernel { /* private fields */ }Expand description
Owned MLX custom Metal kernel function compiled from a source string.
Implementations§
Source§impl MlxMetalKernel
impl MlxMetalKernel
pub fn new( kernel_name: &str, input_names: &[&str], output_names: &[&str], kernel_source: &str, ) -> Result<Self, MlxCError>
pub fn new_with_header( kernel_name: &str, input_names: &[&str], output_names: &[&str], kernel_header: &str, kernel_source: &str, ) -> Result<Self, MlxCError>
pub fn new_with_options( kernel_name: &str, input_names: &[&str], output_names: &[&str], kernel_header: &str, kernel_source: &str, has_atomic_outputs: bool, ) -> Result<Self, MlxCError>
Trait Implementations§
Source§impl Debug for MlxMetalKernel
impl Debug for MlxMetalKernel
Source§impl Drop for MlxMetalKernel
impl Drop for MlxMetalKernel
Auto Trait Implementations§
impl !Send for MlxMetalKernel
impl !Sync for MlxMetalKernel
impl Freeze for MlxMetalKernel
impl RefUnwindSafe for MlxMetalKernel
impl Unpin for MlxMetalKernel
impl UnsafeUnpin for MlxMetalKernel
impl UnwindSafe for MlxMetalKernel
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