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<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
type TileKind = Strided
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: &mut Scope, runtime_args: <RuntimeArgs as RudaType>::ExpandType, config: GlobalReaderConfig, ) -> <Self::Job<EG, NG, ES, NS> as RudaType>::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
type TileKind = Strided
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: &mut Scope, runtime_args: <RuntimeArgs as RudaType>::ExpandType, config: GlobalReaderConfig, ) -> <Self::Job<EG, NG, ES, NS> as RudaType>::ExpandType
const SHOULD_CLEAR: bool = false
Source§impl LaunchArg for RuntimeArgs
impl LaunchArg for RuntimeArgs
Source§type RuntimeArg<R: LaunchMarker> = RuntimeArgsLaunch<R>
type RuntimeArg<R: LaunchMarker> = RuntimeArgsLaunch<R>
The runtime argument for the kernel.
Source§type CompilationArg = RuntimeArgsCompilationArg
type CompilationArg = RuntimeArgsCompilationArg
Compilation argument.
fn register<R: LaunchMarker>( arg: Self::RuntimeArg<R>, launcher: &mut KernelLauncher<R>, ) -> Self::CompilationArg
Source§fn expand(
arg: &Self::CompilationArg,
builder: &mut KernelBuilder,
) -> <Self as RudaType>::ExpandType
fn expand( arg: &Self::CompilationArg, builder: &mut KernelBuilder, ) -> <Self as RudaType>::ExpandType
Register an input variable during compilation that fill the
KernelBuilder.Source§fn expand_output(
arg: &Self::CompilationArg,
builder: &mut KernelBuilder,
) -> <Self as RudaType>::ExpandType
fn expand_output( arg: &Self::CompilationArg, builder: &mut KernelBuilder, ) -> <Self as RudaType>::ExpandType
Register an output variable during compilation that fill the
KernelBuilder.Source§impl RudaType for RuntimeArgs
impl RudaType for RuntimeArgs
type ExpandType = RuntimeArgsExpand
fn __expand_inactive_value(scope: &mut Scope) -> Self::ExpandType
fn __expand_absent_reference( _scope: &mut Scope, _writable: bool, ) -> Self::ExpandType
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> CloneToUninit for Twhere
T: Clone,
impl<T> CloneToUninit for Twhere
T: Clone,
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 Twhere
T: LaunchArg,
impl<T> ViewLayoutLaunchArg for Twhere
T: LaunchArg,
Source§type RuntimeArg<R: LaunchMarker> = <T as LaunchArg>::RuntimeArg<R>
type RuntimeArg<R: LaunchMarker> = <T as LaunchArg>::RuntimeArg<R>
The runtime argument for the kernel.
Source§type CompilationArg = <T as LaunchArg>::CompilationArg
type CompilationArg = <T as LaunchArg>::CompilationArg
Compilation argument.
fn register<R, B>(
arg: <T as ViewLayoutLaunchArg>::RuntimeArg<R>,
_buffer: &B,
_ty: Type,
launcher: &mut KernelLauncher<R>,
) -> <T as ViewLayoutLaunchArg>::CompilationArgwhere
R: LaunchMarker,
B: BufferArg,
Source§fn expand(
arg: &<T as ViewLayoutLaunchArg>::CompilationArg,
_ty: Type,
builder: &mut KernelBuilder,
) -> <T as RudaType>::ExpandType
fn expand( arg: &<T as ViewLayoutLaunchArg>::CompilationArg, _ty: Type, builder: &mut KernelBuilder, ) -> <T as RudaType>::ExpandType
Register an input variable during compilation that fill the
KernelBuilder.Source§fn expand_output(
arg: &<T as ViewLayoutLaunchArg>::CompilationArg,
_ty: Type,
builder: &mut KernelBuilder,
) -> <T as RudaType>::ExpandType
fn expand_output( arg: &<T as ViewLayoutLaunchArg>::CompilationArg, _ty: Type, builder: &mut KernelBuilder, ) -> <T as RudaType>::ExpandType
Register an output variable during compilation that fill the
KernelBuilder.