tenferro-runtime 0.9.0

Concrete tensor helpers, traced tensors, graph execution, and extension runtime support for tenferro.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86

mod cache;
mod cache_owner;
mod capability;
mod engine;
mod engine_registration;
mod error;
mod event_domain;
mod execution;
mod extension;
mod extension_provider;
mod identity;
mod policy;
mod preparation;
mod region;
mod schedule;
mod signature;
mod snapshot;
mod specialization;
mod transfer;

pub use cache::{PreparedPlanCacheLimits, PreparedPlanCacheStats, RuntimeCacheStats};
pub use cache_owner::{
    CacheOwnerError, CacheOwnerFailure, CacheOwnerId, CacheStats, RuntimeCacheError,
    RuntimeCacheOwner, RuntimeStateError,
};
pub use capability::{
    CoreCapabilityBundle, CoreCapabilityBundleBuilder, CoreCapabilityKind, CorePrepareContext,
    DotGeneralPreparation, DotGeneralPrepareRequest, ElementwisePrepareRequest, ElementwiseRuntime,
    ErasedExecutionContext, IndexingPrepareRequest, IndexingRuntime, LayoutPrepareRequest,
    LayoutRuntime, PreparationKeySummary, PrepareCapability, PreparedOperation,
    PreparedOperationBinding, PreparedOperationExecutor, PreparedOperationExecutorHandle,
    PreparedOperationHandle, PreparedOperationPlan, ReductionPrepareRequest, ReductionRuntime,
    UnsupportedReason,
};
pub use engine::{EngineTarget, ExecutionEngine, Output};
pub use engine_registration::{
    assemble_executable_engine_registration, assemble_preparation_only_engine_registration,
    EngineRegistration, EngineRegistrationMetadata, ExecutableEngineRegistrationConfig,
    InputIngressContract, InputPlacementContract, InputSignatureContract,
    PreparationOnlyEngineRegistrationConfig, ResidentOutputContract, RuntimeInputContract,
};
pub(crate) use engine_registration::{ProviderExecutableBinding, ProviderPreparationBinding};
pub use error::{
    EngineExecutionContractError, ExecutionContextMismatch, ExecutionPolicyError,
    ExtensionModuleError, IdentityError, IdentityKind, InputIngressContractError,
    InputSignatureError, InputSpecializationRequirementsError, PlacementConstraintError,
    PrepareError, ProviderContractError, RankRequirement, RegistrationKey, RuntimeConfigError,
    RuntimeReconfigureError, SpecializationError,
};
pub use event_domain::{
    EventDomainDriver, EventDomainError, EventDomainOperation, EventDomainRun, EventToken,
    ImmediateEventDomainDriver,
};
pub use execution::Prepared;
pub use extension::{ExtensionModule, ExtensionModuleId, ExtensionModuleRegistrar};
pub use extension_provider::{ExtensionEngine, ExtensionPlanningConfig, ExtensionPrepareRequest};
pub use identity::{
    EngineId, ExecutionContextIdentity, HardwareClassId, ProviderDeviceIdentity, ProviderId,
    RegistrationIdentity, RuntimeEpoch, RuntimeId,
};
pub use policy::{
    CacheInFlightBehavior, Determinism, ExecutionPolicy, LayoutClass, PrepareOptions,
    PrepareOptionsKey, ProgramPlacementConstraint, ResolvedPlanningConfig, ResolvedPlanningKey,
    ResolvedProgramPlacement, StorageClass, TransferEndpoint,
};
pub use schedule::EventDomainId;
pub use signature::{InputSignature, InputSignatureEntry};
pub use snapshot::{
    EngineSnapshotView, Runtime, RuntimeConfigBuilder, RuntimeConfigSnapshot,
    RuntimeReconfiguration,
};
pub use specialization::{
    InputSpecializationProjection, InputSpecializationRequirements,
    InputSpecializationRequirementsBuilder, LayoutProjection, LayoutSpecialization,
    PlacementProjection, PlacementSpecialization, SpecializationProjection,
    SpecializationRequirements,
};
pub(crate) use transfer::{
    FrozenTransferRegistry, ResolvedTransferEndpoint, ResolvedTransferRoute, TransferRoute,
};
pub use transfer::{
    TransferError, TransferProvider, TransferProviderContractError, TransferRequest,
};

#[cfg(test)]
mod tests;