Skip to main content

veloq_pytorch_query/
lib.rs

1//! Query layer for PyTorch/Kineto traces.
2//!
3//! Each public verb owns a module, matching the NSys query crate style.
4//! Shared request filtering, scope echoing, and wire DTOs stay separate so
5//! verb modules carry only query behavior.
6
7pub mod collectives;
8pub mod correlate;
9pub mod dto;
10pub mod error;
11pub mod filter;
12pub mod inspect;
13pub mod prep;
14mod query_sql;
15pub mod scope;
16pub mod search;
17pub mod slices;
18pub mod stats;
19pub mod summary;
20pub mod time;
21pub mod timeline;
22
23pub use collectives::collectives;
24pub use correlate::correlate;
25pub use dto::{
26    CapabilitiesDto, CaptureFlagsRow, CollectiveRankRow, CollectiveRow, CollectivesAuxiliary,
27    CollectivesResponse, CorrelateResponse, CorrelateRow, EventDetails, EventListAuxiliary,
28    EventRef, InspectResponse, InspectRow, LinkRef, PrepAuxiliary, PrepResponse, PrepRow,
29    SearchResponse, SliceAggregateRow, SliceInstanceRow, SliceRow, SlicesAuxiliary, SlicesResponse,
30    StatsAuxiliary, StatsResponse, StatsRow, SummaryAuxiliary, SummaryResponse, TimeRangeDto,
31    TimelineAuxiliary, TimelineBucketRow, TimelineResponse, TraceFileRow, TraceFileSchemaSurveyDto,
32    TraceSchemaSurveyDto, TypedArgCoverageDto, TypedArgs,
33};
34pub use error::{PytorchQueryError, PytorchQueryResult, SqlPhase};
35pub use filter::{EventFilterRequest, TypeSelection, TypeToken, parse_type_selection};
36pub use inspect::inspect;
37pub use prep::prep_response;
38pub use scope::{PytorchScope, RankScope};
39pub use search::search;
40pub use slices::slices;
41pub use stats::stats;
42pub use summary::summary;
43pub use time::{parse_group_by, resolve_time_window};
44pub use timeline::timeline;