mod allocator_correlation;
mod attempt_domain;
mod backend_close;
mod backend_core;
mod backend_lifecycle;
mod backend_metrics;
pub mod cache;
mod cache_fill;
mod context;
mod deadline;
mod execution_policy;
pub(crate) mod gpu;
mod gpu_execution_pool;
mod io_lifecycle;
mod memory_domain;
mod object_store;
mod observation_completion;
mod process_exit;
mod query_service;
mod reservation;
pub mod resident;
mod service_stream;
#[doc(hidden)]
pub mod source_identity;
mod stream_cleanup;
mod supervised_worker;
#[cfg(test)]
mod tests;
#[doc(hidden)]
pub use attempt_domain::AttemptStreamProviderRoute;
#[cfg(feature = "qualification")]
pub use backend_core::GpuBackendFaultInjector;
pub use backend_core::{
GpuBackendCloseSnapshot, GpuBackendConfig, GpuBackendCore, GpuBackendDeviceConfig,
GpuBackendDeviceInstallationBounds, GpuBackendDeviceProfile, GpuBackendDeviceStartupRecord, GpuBackendSession,
GpuDeviceCapacityProbe, GpuDeviceMemoryLimit, GpuMemoryFraction, GpuMemoryOwnership, ValidatedGpuBackendConfig,
probe_device_capacity,
};
pub(crate) use backend_core::{GpuBackendConstructionCheckpoint, GpuBackendConstructionProbe};
#[cfg(any(test, feature = "test-hooks"))]
pub(crate) use backend_lifecycle::BackendLifecycle;
pub(crate) use backend_lifecycle::ObservationPermit;
pub use backend_metrics::{
BACKEND_ATTEMPT_DURATION_BUCKET_COUNT, BACKEND_METRIC_DEVICE_CAPACITY, BackendAdmissionHealth,
BackendAdmissionMetricsSnapshot, BackendAttemptDisposition, BackendAttemptDuration, BackendAttemptMetricsSnapshot,
BackendAttemptOutcome, BackendDeviceMetricsSnapshot, BackendHistogramProjection, BackendHistogramSnapshot,
BackendMetricAttribute, BackendMetricAttributeValue, BackendMetricProjectionError, BackendMetricSample,
BackendMetricSampleValue, BackendMetricsSnapshot,
};
pub use cache::{
CacheGraphActivity, CacheSourceActivity, CacheSourceEventSnapshot, DeviceCacheReport, cache_observation_fingerprint,
};
pub(crate) use cache_fill::CacheFillByteBudget;
pub use cache_fill::{CacheFillBytePermit, LakehouseCacheFillConfig};
pub use context::{ExecutionCancellation, ExecutionContext};
#[cfg(any(test, feature = "test-hooks"))]
#[doc(hidden)]
pub use context::{RuntimeObservationBuildError, RuntimeObservedExecutionContext, StandaloneConfig};
pub use cudf::io::{KvikioCompatMode, KvikioProcessConfig};
pub use execution_policy::{MemoryResourcePolicy, SourceSchedulerPolicy};
pub(crate) use gpu::{AlgeonGpuInventory, DeviceContext, DeviceLedger, DeviceRegistry, MemoryGrant};
pub use gpu::{
DeviceCapabilities, DeviceCapability, DeviceFacts, DeviceLedgerGrantRow, DeviceLedgerReport,
DeviceMemoryFailureReport, DeviceResourceProfile, DeviceTelemetrySnapshot, DeviceTelemetryStatus,
};
#[doc(hidden)]
pub use gpu_execution_pool::PoolBlockingWork;
pub use gpu_execution_pool::{
GpuExecutionJobKind, GpuExecutionPoolSnapshot, GpuExecutionQuerySnapshot, GpuExecutorLease,
};
pub(crate) use gpu_execution_pool::{GpuExecutionPool, GpuExecutionQueryTracker};
pub use io_lifecycle::IoLifecycle;
#[doc(hidden)]
pub use object_store::S3ObjectStoreKey;
pub use object_store::{S3SourceStorageCredential, S3SourceStorageProfile, SourceStorageProfile};
#[cfg(feature = "qualification")]
pub use query_service::DeviceQuarantineInjection;
#[cfg(any(test, feature = "test-hooks"))]
#[doc(hidden)]
pub use query_service::QueryService;
pub use query_service::{
AdmittedQueryContext, BackendDiagnostics, QueryActivityLease, QueryAdmissionRequest, QueryExecutionConfig,
QueryHandle, QueryServiceAdmissionSnapshot, QueueBounds,
};
#[doc(hidden)]
pub use reservation::RetryFacts;
#[doc(hidden)]
pub use service_stream::AttemptWorkLease;
pub use service_stream::ServiceStreamPool;
pub(crate) use service_stream::{StreamLease, StreamPermitClass};
#[doc(hidden)]
pub use supervised_worker::{SupervisedWorker, SupervisedWorkerJoin};