pub struct AsyncFullStridedLoading {}Expand description
Loads the content of all the stage using all planes, keeping the original layout, making each tile strided
Trait Implementations§
Source§impl Clone for AsyncFullStridedLoading
impl Clone for AsyncFullStridedLoading
Source§fn clone(&self) -> AsyncFullStridedLoading
fn clone(&self) -> AsyncFullStridedLoading
Returns a duplicate of the value. Read more
1.0.0 (const: unstable) · Source§fn clone_from(&mut self, source: &Self)
fn clone_from(&mut self, source: &Self)
Performs copy-assignment from
source. Read moreimpl Copy for AsyncFullStridedLoading
Source§impl CubeType for AsyncFullStridedLoading
impl CubeType for AsyncFullStridedLoading
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 LoadMaxRoundPlaneCount for AsyncFullStridedLoading
impl LoadMaxRoundPlaneCount for AsyncFullStridedLoading
Source§fn max_round_plane_count(
elements_per_tile: u32,
tiles_per_stage: u32,
vector_size: VectorSize,
plane_dim: u32,
dtype: StorageType,
) -> u32
fn max_round_plane_count( elements_per_tile: u32, tiles_per_stage: u32, vector_size: VectorSize, plane_dim: u32, dtype: StorageType, ) -> u32
Returns the largest number of planes that evenly divides the tasks.
Source§impl LoadingValidation for AsyncFullStridedLoading
impl LoadingValidation for AsyncFullStridedLoading
Source§fn validate_with_config(
device_props: &DeviceProperties,
config: &GlobalReaderConfig,
) -> Result<(), InvalidConfigError>
fn validate_with_config( device_props: &DeviceProperties, config: &GlobalReaderConfig, ) -> Result<(), InvalidConfigError>
Verify that configs are valid for a reader, otherwise return an error stating why
fn validate_with_problem( problem: &MatmulProblem, dtypes: &MatmulElems, ident: StageIdent, ) -> Result<(), InvalidConfigError>
Auto Trait Implementations§
impl Freeze for AsyncFullStridedLoading
impl RefUnwindSafe for AsyncFullStridedLoading
impl Send for AsyncFullStridedLoading
impl Sync for AsyncFullStridedLoading
impl Unpin for AsyncFullStridedLoading
impl UnsafeUnpin for AsyncFullStridedLoading
impl UnwindSafe for AsyncFullStridedLoading
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> 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 more