pub struct RuntimeArgs {
pub shape_k: u32,
pub channels: u32,
pub padded_channels: FastDivmod<u32>,
pub operation: ConvolutionOperation,
}Fields§
§shape_k: u32§channels: u32§padded_channels: FastDivmod<u32>§operation: ConvolutionOperationTrait Implementations§
Source§impl Clone for RuntimeArgs
impl Clone for RuntimeArgs
Source§impl CubeType for RuntimeArgs
impl CubeType for RuntimeArgs
Source§impl<TO: TilingOrder> FullLoadingStrategy<RuntimeArgs> for AsyncFullCyclicLoading<TO>
impl<TO: TilingOrder> FullLoadingStrategy<RuntimeArgs> for AsyncFullCyclicLoading<TO>
Source§type TilingLayout = ContiguousTilingLayout<TO>
type TilingLayout = ContiguousTilingLayout<TO>
The layout describing how data is tiled across the stage.
Source§type SyncStrategy = AsyncCopy
type SyncStrategy = AsyncCopy
The synchronization strategy that should be used with this loading strategy
Source§type Job<EG: Numeric, NG: Size, ES: Numeric, NS: Size> = AsyncFullCyclicJob
type Job<EG: Numeric, NG: Size, ES: Numeric, NS: Size> = AsyncFullCyclicJob
The LoadingJob for this strategy.
type Stage = StridedStageFamily
Source§fn new_job<EG: Numeric, NG: Size, ES: Numeric, NS: Size>(
runtime_args: RuntimeArgs,
config: GlobalReaderConfig,
) -> Self::Job<EG, NG, ES, NS>
fn new_job<EG: Numeric, NG: Size, ES: Numeric, NS: Size>( runtime_args: RuntimeArgs, config: GlobalReaderConfig, ) -> Self::Job<EG, NG, ES, NS>
Returns the job with preliminary calculations done.
fn __expand_new_job<EG: Numeric, NG: Size, ES: Numeric, NS: Size>( scope: &Scope, runtime_args: <RuntimeArgs as CubeType>::ExpandType, config: GlobalReaderConfig, ) -> <Self::Job<EG, NG, ES, NS> as CubeType>::ExpandType
const SHOULD_CLEAR: bool = false
Source§impl FullLoadingStrategy<RuntimeArgs> for AsyncFullStridedLoading
impl FullLoadingStrategy<RuntimeArgs> for AsyncFullStridedLoading
Source§type TilingLayout = StridedTilingLayout
type TilingLayout = StridedTilingLayout
The layout describing how data is tiled across the stage.
Source§type SyncStrategy = AsyncCopy
type SyncStrategy = AsyncCopy
The synchronization strategy that should be used with this loading strategy
Source§type Job<EG: Numeric, NG: Size, ES: Numeric, NS: Size> = AsyncFullStridedJob
type Job<EG: Numeric, NG: Size, ES: Numeric, NS: Size> = AsyncFullStridedJob
The LoadingJob for this strategy.
type Stage = StridedStageFamily
Source§fn new_job<EG: Numeric, NG: Size, ES: Numeric, NS: Size>(
runtime_args: RuntimeArgs,
config: GlobalReaderConfig,
) -> Self::Job<EG, NG, ES, NS>
fn new_job<EG: Numeric, NG: Size, ES: Numeric, NS: Size>( runtime_args: RuntimeArgs, config: GlobalReaderConfig, ) -> Self::Job<EG, NG, ES, NS>
Returns the job with preliminary calculations done.
fn __expand_new_job<EG: Numeric, NG: Size, ES: Numeric, NS: Size>( scope: &Scope, runtime_args: <RuntimeArgs as CubeType>::ExpandType, config: GlobalReaderConfig, ) -> <Self::Job<EG, NG, ES, NS> as CubeType>::ExpandType
const SHOULD_CLEAR: bool = false
Source§impl LaunchArg for RuntimeArgs
impl LaunchArg for RuntimeArgs
Source§type RuntimeArg = RuntimeArgsLaunch
type RuntimeArg = RuntimeArgsLaunch
The runtime argument for the kernel.
Source§type CompilationArg = RuntimeArgsCompilationArg
type CompilationArg = RuntimeArgsCompilationArg
Compilation argument.
fn register( arg: Self::RuntimeArg, launcher: &mut KernelLauncher, ) -> Self::CompilationArg
Source§fn expand(
arg: &Self::CompilationArg,
builder: &mut KernelBuilder,
) -> <Self as CubeType>::ExpandType
fn expand( arg: &Self::CompilationArg, builder: &mut KernelBuilder, ) -> <Self as CubeType>::ExpandType
Register a variable during compilation that fill the
KernelBuilder.Auto Trait Implementations§
impl Freeze for RuntimeArgs
impl RefUnwindSafe for RuntimeArgs
impl Send for RuntimeArgs
impl Sync for RuntimeArgs
impl Unpin for RuntimeArgs
impl UnsafeUnpin for RuntimeArgs
impl UnwindSafe for RuntimeArgs
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
impl<ST, DT> CastableFrom<ST, Initialized, Initialized> for DT
impl<ST, DT> CastableFrom<ST, Uninit, Uninit> for DT
Source§impl<T> CloneExpand for Twhere
T: Clone,
impl<T> CloneExpand for Twhere
T: Clone,
fn __expand_clone_method(&self, _: &Scope) -> T
Source§impl<T> CloneToUninit for Twhere
T: Clone,
impl<T> CloneToUninit for Twhere
T: Clone,
Source§impl<T> Downcast for Twhere
T: Any,
impl<T> Downcast for Twhere
T: Any,
Source§fn into_any(self: Box<T>) -> Box<dyn Any>
fn into_any(self: Box<T>) -> Box<dyn Any>
Converts
Box<dyn Trait> (where Trait: Downcast) to Box<dyn Any>, which can then be
downcast into Box<dyn ConcreteType> where ConcreteType implements Trait.Source§fn into_any_rc(self: Rc<T>) -> Rc<dyn Any>
fn into_any_rc(self: Rc<T>) -> Rc<dyn Any>
Converts
Rc<Trait> (where Trait: Downcast) to Rc<Any>, which can then be further
downcast into Rc<ConcreteType> where ConcreteType implements Trait.Source§fn as_any(&self) -> &(dyn Any + 'static)
fn as_any(&self) -> &(dyn Any + 'static)
Converts
&Trait (where Trait: Downcast) to &Any. This is needed since Rust cannot
generate &Any’s vtable from &Trait’s.Source§fn as_any_mut(&mut self) -> &mut (dyn Any + 'static)
fn as_any_mut(&mut self) -> &mut (dyn Any + 'static)
Converts
&mut Trait (where Trait: Downcast) to &Any. This is needed since Rust cannot
generate &mut Any’s vtable from &mut Trait’s.Source§impl<T> DowncastSend for T
impl<T> DowncastSend for T
Source§impl<T> DowncastSync for T
impl<T> DowncastSync for T
impl<T> ErasedDestructor for Twhere
T: 'static,
Source§impl<T> Instrument for T
impl<T> Instrument for T
Source§fn instrument(self, span: Span) -> Instrumented<Self> ⓘ
fn instrument(self, span: Span) -> Instrumented<Self> ⓘ
Source§fn in_current_span(self) -> Instrumented<Self> ⓘ
fn in_current_span(self) -> Instrumented<Self> ⓘ
Source§impl<T> IntoEither for T
impl<T> IntoEither for T
Source§fn into_either(self, into_left: bool) -> Either<Self, Self> ⓘ
fn into_either(self, into_left: bool) -> Either<Self, Self> ⓘ
Converts
self into a Left variant of Either<Self, Self>
if into_left is true.
Converts self into a Right variant of Either<Self, Self>
otherwise. Read moreSource§fn into_either_with<F>(self, into_left: F) -> Either<Self, Self> ⓘ
fn into_either_with<F>(self, into_left: F) -> Either<Self, Self> ⓘ
Converts
self into a Left variant of Either<Self, Self>
if into_left(&self) returns true.
Converts self into a Right variant of Either<Self, Self>
otherwise. Read moreSource§impl<T> Pointable for T
impl<T> Pointable for T
impl<T> Read<Exclusive, BecauseExclusive> for Twhere
T: ?Sized,
impl<T> RuntimeConfig for T
Source§impl<T> TuneInputs for T
impl<T> TuneInputs for T
Source§impl<T> ViewLayoutLaunchArg for T
impl<T> ViewLayoutLaunchArg for T
Source§type RuntimeArg = <T as LaunchArg>::RuntimeArg
type RuntimeArg = <T as LaunchArg>::RuntimeArg
The runtime argument for the kernel.
Source§type CompilationArg = <T as LaunchArg>::CompilationArg
type CompilationArg = <T as LaunchArg>::CompilationArg
Compilation argument.
fn register<B>(
arg: <T as ViewLayoutLaunchArg>::RuntimeArg,
_buffer: &B,
_ty: Type,
launcher: &mut KernelLauncher,
) -> <T as ViewLayoutLaunchArg>::CompilationArgwhere
B: MemoryArg,
Source§fn expand(
arg: &<T as ViewLayoutLaunchArg>::CompilationArg,
_ty: Type,
builder: &mut KernelBuilder,
) -> <T as CubeType>::ExpandType
fn expand( arg: &<T as ViewLayoutLaunchArg>::CompilationArg, _ty: Type, builder: &mut KernelBuilder, ) -> <T as CubeType>::ExpandType
Register an input variable during compilation that fill the
KernelBuilder.Source§fn expand_output(
arg: &Self::CompilationArg,
ty: Type,
builder: &mut KernelBuilder,
) -> Self::ExpandType
fn expand_output( arg: &Self::CompilationArg, ty: Type, builder: &mut KernelBuilder, ) -> Self::ExpandType
Register an output variable during compilation that fill the
KernelBuilder.