xlog_cuda/
lib.rs

1//! GPU kernel provider for XLOG
2
3pub mod arrow_device;
4pub mod cuda_compat;
5pub mod cuda_graph;
6pub mod device;
7pub mod device_pool;
8pub mod device_runtime;
9pub mod dlpack;
10pub mod joint_constraint;
11pub mod joint_solver;
12pub mod kernel_manifest_data;
13pub mod launch;
14pub mod memory;
15pub mod multi_gpu_memory;
16pub mod provider;
17mod semantic_hypergraph;
18mod semantic_program;
19mod semantic_training_view;
20mod semantic_transition;
21mod semantic_work;
22pub mod type_seam;
23pub mod wcoj_metadata;
24#[cfg(feature = "wcoj-phase-timing")]
25pub mod wcoj_phase_timing;
26
27pub(crate) mod embedded_kernel_data {
28    include!(concat!(env!("OUT_DIR"), "/embedded_kernel_data.rs"));
29}
30
31pub use arrow_device::{ArrowDeviceArray, ArrowDeviceArrayOwned, ARROW_DEVICE_CUDA};
32pub use cuda_compat::{
33    sys, AsKernelParam, CudaFunction, CudaSlice, CudaStream, CudaView, CudaViewMut, DevicePtr,
34    DevicePtrMut, DeviceRepr, DeviceSlice, DriverError, IntoKernelParamStorage, KernelParamStorage,
35    KernelScalar, LaunchAsync, LaunchConfig, ValidAsZeroBits,
36};
37pub use device::CudaDevice;
38pub use device_pool::GpuDevicePool;
39pub use dlpack::{DLManagedTensor, DlpackManagedTensor, DlpackTable};
40pub use joint_constraint::{CarrierBufferId, CarrierError, CarrierExport, JointConstraintCarrier};
41pub use joint_solver::{
42    Component, ConstraintGraph, Feasibility, FuelMeter, SolveStrategy, SolverError,
43    SOLVER_ABI_IDENTITY,
44};
45pub use memory::{CudaBuffer, CudaColumn, GpuMemoryManager, RuntimeAllocBlock};
46pub use multi_gpu_memory::MultiGpuMemoryManager;
47pub use provider::{
48    circuit_kernels, dedup_kernels, filter_kernels, groupby_kernels, ilp_kernels, join_kernels,
49    pack_kernels, pir_kernels, scan_kernels, set_ops_kernels, sort_kernels, CompareOp,
50    CudaKernelProvider, CudaProviderBuilder, FjDeltaCols, FjNode, FjPlan, FjSubAtom,
51    IlpExactNaryPatterns, IlpExactNaryRequest, JoinIndexV2, JoinType, CIRCUIT_MODULE, DEDUP_MODULE,
52    GROUPBY_MODULE, ILP_MODULE, JOIN_MODULE, PACK_MODULE, PIR_MODULE, SCAN_MODULE, SET_OPS_MODULE,
53    SORT_MODULE,
54};
55pub use semantic_hypergraph::{
56    SemanticAdmission, SemanticAdmissionLimits, SemanticAdmissionRecords, SemanticArgument,
57    SemanticExtents, SemanticForkHandle, SemanticHandleKind, SemanticHypergraph,
58    SemanticHypergraphCapacities, SemanticHypergraphError, SemanticHypergraphExecutionStats,
59    SemanticInsertOutcome, SemanticInsertedSupport, SemanticPolarity, SemanticPredicateRecord,
60    SemanticRecordRole, SemanticRootDigest, SemanticRootHandle, SemanticRootSnapshot,
61    SemanticStatementHandle, SemanticStatementIdentity, SemanticStatementKey, SemanticStatementRef,
62    SemanticSupportEvent, SemanticSupportHandle, SemanticSupportIdentity, SemanticSupportRecord,
63    SemanticSupportRef, SemanticTruth, SemanticTypedRecord, SemanticVersionHandle,
64    SemanticVersionIdentity, SemanticVersionRef, SemanticView,
65};
66pub use semantic_program::{SemanticProgramAdmission, SemanticProgramFact, SemanticProgramRule};
67pub use semantic_training_view::{
68    validate_training_view_layout, SemanticSelectedTrainingView, SemanticTrainingCanary,
69    SemanticTrainingCanaryKind, SemanticTrainingCanaryRecord, SemanticTrainingCanaryRefusalReason,
70    SemanticTrainingCanaryRefusalRecord, SemanticTrainingDomain, SemanticTrainingManifest,
71    SemanticTrainingObjective, SemanticTrainingObjectiveGroup, SemanticTrainingObjectiveGroupKind,
72    SemanticTrainingObjectiveGroupRecord, SemanticTrainingObjectiveRecord,
73    SemanticTrainingRosterRow, SemanticTrainingViewBasis, SemanticTrainingViewLayout,
74    SemanticTrainingViewOrigin, SemanticTrainingViewOriginRecord, SemanticTrainingViewPort,
75    SemanticTrainingViewRow, SemanticTrainingViewSelection, SEMANTIC_TRAINING_CANARY_EVALUATOR_ABI,
76};
77pub use semantic_transition::{
78    initial_source_layout, output_intents_from_publication_material,
79    pending_output_intents_from_publication_material, semantic_structural_cost_descriptor,
80    Identity256, SemanticActionCatalogue, SemanticActionDescriptor, SemanticActiveRow,
81    SemanticActiveRows, SemanticCatalogueBinding, SemanticCatalogueField,
82    SemanticCatalogueSignature, SemanticCheckpointNativeProjection, SemanticComponent,
83    SemanticComponentInput, SemanticContinuationInput, SemanticFeedback,
84    SemanticFeedbackRecordMaterial, SemanticFeedbackSchema, SemanticGradientDeliveryBinding,
85    SemanticInitialPrefillContentWitness, SemanticInitialPrefillLease, SemanticInitialSourceLayout,
86    SemanticLearningCopyReset, SemanticLearningPhase, SemanticLearningPhaseRecord,
87    SemanticLearningPhaseTransition, SemanticModelContext, SemanticModelContractLayout,
88    SemanticModelForwardWitness, SemanticModelMemory, SemanticModelStorage, SemanticModelView,
89    SemanticObservedSource, SemanticOutputIntent, SemanticParentBinding, SemanticPolicyFieldLayout,
90    SemanticPolicyLayout, SemanticPreparedStep, SemanticPreparedStepOutcome,
91    SemanticPublicationInputProjection, SemanticPublicationTensorProjection,
92    SemanticPublishedIdentity, SemanticPublishedLease, SemanticReplayMaterial,
93    SemanticResidentModelMemory, SemanticRngBinding, SemanticSegmentColdCapacity,
94    SemanticSourceMapping, SemanticStateRecord, SemanticStateRole,
95    SemanticStructuralCostDescriptor, SemanticTaskContentIdentity, SemanticTaskEvaluation,
96    SemanticTaskEvaluationSpec, SemanticTaskFacts, SemanticTaskGoalWitness,
97    SemanticTaskObjectiveLaw, SemanticTaskObservation, SemanticTaskObservationRoots,
98    SemanticTaskPriorityGoal, SemanticTaskPriorityLevel, SemanticTaskProgram, SemanticTaskRefusal,
99    SemanticTaskScoring, SemanticTensorContentWitness, SemanticTensorInput, SemanticTensorLayout,
100    SemanticTextRow, SemanticTextSlot, SemanticTokenProvenance, SemanticTransitionError,
101    SemanticTransitionHostIoStats, SemanticTransitionKind, SemanticTransitionLane,
102    SemanticTransitionObservation, SemanticTransitionOutcome, SemanticTransitionReceipt,
103    SemanticTransitionRefusal, SemanticTransitionSession, SemanticTransitionWork,
104    SEMANTIC_FEEDBACK_SUPPORT_FIELDS, SEMANTIC_TRANSITION_COMPONENT_COUNT,
105    SEMANTIC_TRANSITION_GENERATION,
106};
107#[cfg(feature = "semantic-policy")]
108pub use semantic_transition::{
109    SemanticCompletedActionComponentMaterial, SemanticCompletedActionLaneMaterial,
110    SemanticCompletedActionProjectionMaterial, SemanticCompletedEditSolutionMaterial,
111    SemanticCompletedExecutionObservation, SemanticCompletedLaneOutcomeMaterial,
112    SemanticCompletedMaterialCapacity, SemanticCompletedModelCarrierMaterial,
113    SemanticCompletedReplayMaterials, SemanticCompletedStepWitnessMaterial,
114    SemanticCompletedTaskGroundMaterial, SemanticCompletedTheoryDeltaMaterial,
115    SemanticPolicyGradients, SemanticSelectedPolicyGradients, SemanticUpdateCanaryMeasurement,
116    SemanticUpdateForwardReceipt, SemanticUpdateForwardReceiptMaterial, SemanticUpdateMeasurements,
117};
118#[cfg(feature = "semantic-policy")]
119pub use semantic_transition::{
120    SemanticCompletedModelEvaluation, SemanticEvaluationCohort, SemanticModelEvaluation,
121    SemanticModelEvaluationResult,
122};
123pub use semantic_work::ModelWorkKind;
124
125pub use wcoj_metadata::{
126    HeatDist, LayoutSignature, RootMetadata, VertexId, WcojCycle4HgWorkPlanU32,
127    WcojCycle4HgWorkPlanU64, WcojRelationMetadata, WcojTriangleHgWorkPlanU32,
128    WcojTriangleHgWorkPlanU64,
129};
xlog_cuda/lib.rs

xlog_cuda/
lib.rs
