Expand description
Local sys-level wrappers around cudarc::cutensor::sys for the
cuTENSOR entry points the safe cudarc::cutensor::result module
does not expose (Reduce/ElementwiseBinary/ElementwiseTrinary
create+execute, Permutation create+execute, predefined compute
descriptors).
Every function here is unsafe and takes the raw cuTENSOR enum
types from cudarc::cutensor::sys. The entire crate’s actor layer
drives these through kernel/tensor/.
All wrappers convert cutensorStatus_t into a thin
CutensorError that mirrors what cudarc::cutensor::result
emits; callers that already use cudarc::cutensor::result can
interleave both freely.
Structs§
- Cutensor
Error - Error wrapper around a
cutensorStatus_t. Mirrorscudarc::cutensor::result::CutensorErrorso error messages are consistent across the safe/sys boundary.
Functions§
- c_32f
- create_
elementwise_ ⚠binary - Create a binary elementwise operation descriptor. Wraps
cutensorCreateElementwiseBinary. - create_
elementwise_ ⚠trinary - Create a trinary elementwise operation descriptor. Wraps
cutensorCreateElementwiseTrinary. - create_
permutation ⚠ - Create a permutation operation descriptor. Wraps
cutensorCreatePermutation. - elementwise_
binary_ ⚠execute - Execute a previously-planned binary elementwise op.
- elementwise_
trinary_ ⚠execute - Execute a previously-planned trinary elementwise op.
- permute⚠
- Execute a previously-planned permutation.
- plan_
preference_ ⚠set_ algo - Set the pinned algorithm on a plan-preference object. Used by the
contraction autotune to probe a specific
cutensorAlgo_tvalue. - r_32f
- r_64f
- r_
min_ 16bf - r_
min_ 16f - r_
min_ 32f - r_
min_ 64f - r_
min_ tf32 - reduce⚠
- Wraps
cutensorReduce(post-plan execution).