Skip to main content

Crate sim_lib_compute_rocm

Crate sim_lib_compute_rocm 

Source
Expand description

Optional runtime-loaded ROCm/rocBLAS tensor compute provider.

The ROCm provider has no link-time ROCm dependency. It validates HIP and rocBLAS entry points plus an observed AMD gfx* target at runtime and exports a tensor site only after that ABI evidence exists. The executor accepts dense matmul for f32 plus half-family dtypes when rocBLASLt evidence is present.

Structs§

ComputeRocmLib
Loadable library that registers a ROCm compute site when runtime validation succeeds.
FakeRocmLoader
Fake loader for deterministic tests.
RocmAbiEvidence
Dynamic-library ABI evidence required by the ROCm provider.
RocmAllocation
Opaque ROCm allocation evidence.
RocmLibrarySet
Loaded ROCm runtime libraries kept alive for function-pointer validity.
RocmLoadError
ROCm dynamic-loading failure.
RocmResidentStorage
Resident storage for a ROCm matmul result.
RocmRuntimeLoader
Real dynamic loader using platform ROCm shared libraries.
RocmRuntimeProbe
Result of ROCm runtime discovery.
RocmSymbolEvidence
One validated runtime symbol.
RocmTensorExecutor
Tensor executor backed by validated ROCm/rocBLAS ABI evidence.

Statics§

RECIPES
Cookbook recipes for this lib, embedded at build time.

Traits§

DynamicRocmLoader
Loader abstraction used by real and fake ROCm discovery.

Functions§

compute_rocm_capability
Capability required before a ROCm hardware tensor site can be realized.
compute_rocm_lib_symbol
Stable symbol for the ROCm runtime library.
compute_rocm_site_symbol
Site symbol exported by a successful ROCm runtime.
discover_rocm_runtime
Discovers ROCm using the real platform dynamic loader.
rocm_executor_symbol
Stable symbol for the ROCm tensor executor.