query_runtime/
runtime.rs

1//! Runtime, memory, boundary, and reservation contracts for native execution.
2//!
3//! Values defined in this module govern observable execution behavior across crates.
4//! Policy enum string representations, reservation fact keys, and retry fact keys
5//! are exposed to tests, telemetry reports, and adapter diagnostics.
6//! These identifiers constitute stable diagnostic contracts and are updated
7//! in coordination with reporting schemas.
8
9mod allocator_correlation;
10mod attempt_domain;
11mod backend_close;
12mod backend_core;
13mod backend_lifecycle;
14mod backend_metrics;
15pub mod cache;
16mod cache_fill;
17mod context;
18mod deadline;
19mod execution_policy;
20pub(crate) mod gpu;
21mod gpu_execution_pool;
22mod io_lifecycle;
23mod memory_domain;
24mod object_store;
25mod observation_completion;
26mod process_exit;
27mod query_service;
28mod reservation;
29/// Frontend-neutral GPU-resident data and reusable build handles.
30pub mod resident;
31mod service_stream;
32#[doc(hidden)]
33pub mod source_identity;
34mod stream_cleanup;
35mod supervised_worker;
36#[cfg(test)]
37mod tests;
38
39#[doc(hidden)]
40pub use attempt_domain::AttemptStreamProviderRoute;
41#[cfg(feature = "qualification")]
42pub use backend_core::GpuBackendFaultInjector;
43pub use backend_core::{
44    GpuBackendCloseSnapshot, GpuBackendConfig, GpuBackendCore, GpuBackendDeviceConfig,
45    GpuBackendDeviceInstallationBounds, GpuBackendDeviceProfile, GpuBackendDeviceStartupRecord, GpuBackendSession,
46    GpuDeviceCapacityProbe, GpuDeviceMemoryLimit, GpuMemoryFraction, GpuMemoryOwnership, ValidatedGpuBackendConfig,
47    probe_device_capacity,
48};
49pub(crate) use backend_core::{GpuBackendConstructionCheckpoint, GpuBackendConstructionProbe};
50#[cfg(any(test, feature = "test-hooks"))]
51pub(crate) use backend_lifecycle::BackendLifecycle;
52pub(crate) use backend_lifecycle::ObservationPermit;
53pub use backend_metrics::{
54    BACKEND_ATTEMPT_DURATION_BUCKET_COUNT, BACKEND_METRIC_DEVICE_CAPACITY, BackendAdmissionHealth,
55    BackendAdmissionMetricsSnapshot, BackendAttemptDisposition, BackendAttemptDuration, BackendAttemptMetricsSnapshot,
56    BackendAttemptOutcome, BackendDeviceMetricsSnapshot, BackendHistogramProjection, BackendHistogramSnapshot,
57    BackendMetricAttribute, BackendMetricAttributeValue, BackendMetricProjectionError, BackendMetricSample,
58    BackendMetricSampleValue, BackendMetricsSnapshot,
59};
60pub use cache::{
61    CacheGraphActivity, CacheSourceActivity, CacheSourceEventSnapshot, DeviceCacheReport, cache_observation_fingerprint,
62};
63pub(crate) use cache_fill::CacheFillByteBudget;
64pub use cache_fill::{CacheFillBytePermit, LakehouseCacheFillConfig};
65pub use context::{ExecutionCancellation, ExecutionContext};
66#[cfg(any(test, feature = "test-hooks"))]
67#[doc(hidden)]
68pub use context::{RuntimeObservationBuildError, RuntimeObservedExecutionContext, StandaloneConfig};
69pub use cudf::io::{KvikioCompatMode, KvikioProcessConfig};
70pub use execution_policy::{MemoryResourcePolicy, SourceSchedulerPolicy};
71pub(crate) use gpu::{AlgeonGpuInventory, DeviceContext, DeviceLedger, DeviceRegistry, MemoryGrant};
72pub use gpu::{
73    DeviceCapabilities, DeviceCapability, DeviceFacts, DeviceLedgerGrantRow, DeviceLedgerReport,
74    DeviceMemoryFailureReport, DeviceResourceProfile, DeviceTelemetrySnapshot, DeviceTelemetryStatus,
75};
76#[doc(hidden)]
77pub use gpu_execution_pool::PoolBlockingWork;
78pub use gpu_execution_pool::{
79    GpuExecutionJobKind, GpuExecutionPoolSnapshot, GpuExecutionQuerySnapshot, GpuExecutorLease,
80};
81pub(crate) use gpu_execution_pool::{GpuExecutionPool, GpuExecutionQueryTracker};
82pub use io_lifecycle::IoLifecycle;
83#[doc(hidden)]
84pub use object_store::S3ObjectStoreKey;
85pub use object_store::{S3SourceStorageCredential, S3SourceStorageProfile, SourceStorageProfile};
86#[cfg(feature = "qualification")]
87pub use query_service::DeviceQuarantineInjection;
88#[cfg(any(test, feature = "test-hooks"))]
89#[doc(hidden)]
90pub use query_service::QueryService;
91pub use query_service::{
92    AdmittedQueryContext, BackendDiagnostics, QueryActivityLease, QueryAdmissionRequest, QueryExecutionConfig,
93    QueryHandle, QueryServiceAdmissionSnapshot, QueueBounds,
94};
95#[doc(hidden)]
96pub use reservation::RetryFacts;
97#[doc(hidden)]
98pub use service_stream::AttemptWorkLease;
99pub use service_stream::ServiceStreamPool;
100pub(crate) use service_stream::{StreamLease, StreamPermitClass};
101#[doc(hidden)]
102pub use supervised_worker::{SupervisedWorker, SupervisedWorkerJoin};
query_runtime/runtime.rs

query_runtime/
runtime.rs
