sim_lib_compute_rocm/
lib.rs

1#![allow(unsafe_code)]
2#![deny(missing_docs)]
3//! Optional runtime-loaded ROCm/rocBLAS tensor compute provider.
4//!
5//! The ROCm provider has no link-time ROCm dependency. It validates HIP and
6//! rocBLAS entry points plus an observed AMD `gfx*` target at runtime and
7//! exports a tensor site only while the live library handles remain retained.
8//! Dense `f32` matmul executes through rocBLAS and retains the result in device
9//! storage. Half-family requests fail closed until a rocBLASLt execution path
10//! exists.
11
12mod loader;
13mod runtime;
14mod site;
15mod storage;
16
17pub use loader::{
18    DynamicRocmLoader, FakeRocmLoader, RocmAbiEvidence, RocmLibrarySet, RocmLoadError,
19    RocmProbePort, RocmRuntimeLoader, RocmRuntimeProbe, RocmSymbolEvidence, discover_rocm_runtime,
20};
21pub use site::{
22    ComputeRocmLib, RocmTensorExecutor, compute_rocm_capability, compute_rocm_lib_symbol,
23    compute_rocm_site_symbol, rocm_executor_symbol,
24};
25pub use storage::{RocmAllocation, RocmResidentStorage};
26
27/// Cookbook recipes for this lib, embedded at build time.
28pub static RECIPES: sim_cookbook::EmbeddedDir =
29    include!(concat!(env!("OUT_DIR"), "/cookbook_recipes.rs"));
30
31#[cfg(test)]
32mod tests;
sim_lib_compute_rocm/lib.rs

sim_lib_compute_rocm/
lib.rs
