pub struct BlockScaledLayoutLaunch<'a, R: Runtime> { /* private fields */ }Implementations§
Source§impl<'a, R: Runtime> BlockScaledLayoutLaunch<'a, R>
impl<'a, R: Runtime> BlockScaledLayoutLaunch<'a, R>
Sourcepub fn new(
tensor_shape: <Sequence<FastDivmod> as LaunchArg>::RuntimeArg<'a, R>,
tensor_len: <u32 as LaunchArg>::RuntimeArg<'a, R>,
scales_strides: <Sequence<u32> as LaunchArg>::RuntimeArg<'a, R>,
block_size: Vec<u8>,
scales_line_size: u32,
) -> Self
pub fn new( tensor_shape: <Sequence<FastDivmod> as LaunchArg>::RuntimeArg<'a, R>, tensor_len: <u32 as LaunchArg>::RuntimeArg<'a, R>, scales_strides: <Sequence<u32> as LaunchArg>::RuntimeArg<'a, R>, block_size: Vec<u8>, scales_line_size: u32, ) -> Self
New kernel
Trait Implementations§
Source§impl<'a, R: Runtime> ArgSettings<R> for BlockScaledLayoutLaunch<'a, R>
impl<'a, R: Runtime> ArgSettings<R> for BlockScaledLayoutLaunch<'a, R>
Source§fn register(&self, launcher: &mut KernelLauncher<R>)
fn register(&self, launcher: &mut KernelLauncher<R>)
Register the information of an argument to the KernelLauncher.
Auto Trait Implementations§
impl<'a, R> Freeze for BlockScaledLayoutLaunch<'a, R>
impl<'a, R> RefUnwindSafe for BlockScaledLayoutLaunch<'a, R>where
R: RefUnwindSafe,
impl<'a, R> Send for BlockScaledLayoutLaunch<'a, R>
impl<'a, R> Sync for BlockScaledLayoutLaunch<'a, R>
impl<'a, R> Unpin for BlockScaledLayoutLaunch<'a, R>where
R: Unpin,
impl<'a, R> UnwindSafe for BlockScaledLayoutLaunch<'a, R>where
R: UnwindSafe,
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