Expand description
CPU backend computation kernels for fuel.
This crate contains the core CPU computation helpers (MatMul, pooling,
convolution, reductions, etc.) extracted from fuel-core so they can be
reused and tested independently. It also re-exports the MKL and Accelerate
FFI bindings when the corresponding features are enabled.
Re-exports§
pub use byte_storage::CPU_ALIGN_BYTES;pub use byte_storage::CpuStorageBytes;pub use dyn_impl::CpuStorage;pub use quantized::CpuQStorage;pub use ops::*;pub use utils::*;
Modules§
- byte_
kernels - Typed byte-shaped kernels — Phase 7.5 B5. These operate on
CpuStorageBytesdirectly viabytemucktyped views. The dispatch wrapper infuel_dispatch::dispatch::cpu_wrappersextractsCpuStorageBytesfromBackendStorage::Cpu(...)and calls these kernels. Typed byte-shaped kernels — Phase 7.5 B5. - byte_
storage - Phase 7.5 storage-unification target: byte-shaped CPU storage
with
Arc<AlignedBytes>backing, 64-byte alignment, and CoW mutation. Coexists with the legacyCpuStorage(HostBuffer-based) during op-kernel migration; legacy retires when the last kernel migrates. Byte-shaped CPU storage — Phase 7.5 storage-unification target. - chassis
- Trait-chassis surface shared by elementwise + reduction CPU
kernels. One shape/loop pass per kernel family; op-specific math
lives in tiny trait impls. See
chassis::reduction. Trait-chassis surface for CPU kernels. - conv2d
- dyn_
impl DynBackendStorageandDynBackendDeviceimplementations for the CPU backend.- host_
storage - Alternative
HostStorageimpls beyond the default ownedcrate::dyn_impl::CpuBackendStorage(crate::dyn_impl:: CpuBackendStorage). - ops
- CPU backend operation helper structs and functions.
- probe
- Device enumeration for the default CPU backend.
- quantized
- CPU-side quantized storage adapter — bridges
Box<dyn QuantizedType>(per-block-format scalar ops from fuel-quantized) to the backend-agnosticDynQuantizedStoragetrait that fuel-core dispatches against. - system_
memory - OS-level system RAM queries for the CPU backend’s
fuel_backend_contract::backend::BackendRuntimeimpl. - utils