Skip to main content

Module compute_desc

Module compute_desc 

Source
Expand description

Mixed-precision compute descriptors used by every cuTENSOR op.

cuTENSOR’s cutensorComputeDescriptor_t is opaque; the library exports a fixed set of named globals (CUTENSOR_R_MIN_32F, …). We expose them as a Rust enum and resolve to the underlying pointer at the call site.

Enums§

ComputeDesc
Selects the cuTENSOR mixed-precision compute path. MinF32 is the natural pairing for f16/bf16 inputs that accumulate in f32.

Functions§

compute_desc_tag
Stable u32 fingerprint of a ComputeDesc for plan-cache keys. Different from tag() so two compute descs are guaranteed to hash distinctly without relying on string hashing.
resolve_compute_desc
Resolve a ComputeDesc to the corresponding extern global pointer cuTENSOR expects.