sim_lib_compute_wgpu/
lib.rs

1#![forbid(unsafe_code)]
2#![deny(missing_docs)]
3//! Evidence-based `wgpu` tensor compute site discovery.
4//!
5//! This crate discovers physical adapters deterministically, requests a bounded
6//! device profile, records granted limits and features, and exports SIM tensor
7//! sites only for adapters that pass transfer, mapping, and allocation probes.
8
9mod arena;
10mod counters;
11mod dispatch;
12mod dispatch_linalg;
13mod dispatch_reductions;
14mod kernel_elementwise;
15mod kernel_linalg;
16mod kernel_reductions;
17mod kernel_support;
18mod kernel_wgsl;
19mod kernels;
20mod materialization;
21mod pipeline;
22mod probe;
23mod queue;
24mod segments;
25mod site;
26mod storage;
27mod transfer;
28
29pub use arena::{WgpuAllocationId, WgpuArenaAllocation, WgpuArenaSnapshot, WgpuResidentArena};
30pub(crate) use counters::WgpuPhysicalCounters;
31pub use counters::{BufferRange, PhysicalSubmissionEvidence};
32pub use materialization::WgpuMaterializationCache;
33pub use pipeline::{
34    WgpuKernelDType, WgpuKernelOp, WgpuPipelineCache, WgpuPipelineCacheSnapshot, WgpuPipelineKey,
35    WgpuPipelineRecord, WgpuTileProfile,
36};
37pub use probe::{
38    AllocationAttempt, ProbeEvidence, ProbePolicy, RequestedWgpuProfile, TransferEvidence,
39    WgpuAdapterEvidence, WgpuAdapterProbe, WgpuCapabilityEvidence, WgpuDiscovery,
40    WgpuDiscoveryError, WgpuLimitEvidence, discover_wgpu_adapters,
41};
42pub use queue::{WgpuQueueLimits, WgpuQueueSnapshot, WgpuSubmissionQueue};
43pub use segments::{WgpuResidentSegment, WgpuSegmentPlan};
44pub use site::{
45    ComputeWgpuLib, WgpuTensorExecutor, compute_wgpu_capability, compute_wgpu_lib_symbol,
46    compute_wgpu_site_symbol, wgpu_executor_symbol,
47};
48pub use storage::WgpuResidentStorage;
49pub(crate) use storage::WgpuResidentStorageDescriptor;
50pub use transfer::{WgpuTransferPlan, WgpuTransferSpan};
51
52/// Cookbook recipes for this lib, embedded at build time.
53pub static RECIPES: sim_cookbook::EmbeddedDir =
54    include!(concat!(env!("OUT_DIR"), "/cookbook_recipes.rs"));
55
56#[cfg(test)]
57mod tests;
sim_lib_compute_wgpu/lib.rs

sim_lib_compute_wgpu/
lib.rs
