Expand description
Raw FFI + dynamic loader skeleton for NVIDIA cuTENSOR.
baracuda-cutensor wraps this with a safe, typed API. Use this
crate directly only if you need a function that the safe layer
hasn’t wrapped yet (in which case please file a bug).
cuTENSOR is a separately-installed NVIDIA library for high-performance tensor contraction, reduction, and element-wise ops. v0.1 ships the loader + status enum; concrete contraction/permutation/reduction wrappers follow once CI has a cuTENSOR install.
Modules§
- cutensor
Algo cutensorAlgo_t— algorithm selector for contraction planning.- cutensor
Data Type cutensorDataType_t— element type enum.- cutensor
JitMode cutensorJitMode_t— Just-in-time-compile selector (cuTENSOR 2.x).- cutensor
Operator cutensorOperator_t— element-wise op selector.- cutensor
Worksize Preference cutensorWorksizePreference_t.
Structs§
- Cutensor
- Lazily-resolved cuTENSOR function-pointer table.
- cutensor
Status_ t - cuTENSOR status code.
Functions§
- cutensor
- Return the lazily-loaded cuTENSOR library accessor.
Type Aliases§
- PFN_
cutensor Block Sparse Contract - Function-pointer type for
cutensorBlockSparseContract(execute block-sparse tensor contraction). See https://docs.nvidia.com/cuda/cutensor/index.html. - PFN_
cutensor Compute Descriptor GetAttribute - Function-pointer type for
cutensorComputeDescriptorGetAttribute(get an attribute on a compute descriptor). See https://docs.nvidia.com/cuda/cutensor/index.html. - PFN_
cutensor Compute Descriptor SetAttribute - Function-pointer type for
cutensorComputeDescriptorSetAttribute(set an attribute on a compute descriptor). See https://docs.nvidia.com/cuda/cutensor/index.html. - PFN_
cutensor Contract - Function-pointer type for
cutensorContract(execute tensor contraction). See https://docs.nvidia.com/cuda/cutensor/index.html. - PFN_
cutensor Contract Trinary - Function-pointer type for
cutensorContractTrinary(execute three-tensor contraction). See https://docs.nvidia.com/cuda/cutensor/index.html. - PFN_
cutensor Create - Function-pointer type for
cutensorCreate(create cuTENSOR library handle). See https://docs.nvidia.com/cuda/cutensor/index.html. - PFN_
cutensor Create Block Sparse Contraction - Function-pointer type for
cutensorCreateBlockSparseContraction(build an operation descriptor for a block-sparse contraction). See https://docs.nvidia.com/cuda/cutensor/index.html. - PFN_
cutensor Create Block Sparse Tensor Descriptor - Function-pointer type for
cutensorCreateBlockSparseTensorDescriptor(create a block-sparse tensor descriptor). See https://docs.nvidia.com/cuda/cutensor/index.html. - PFN_
cutensor Create Compute Descriptor - Function-pointer type for
cutensorCreateComputeDescriptor(create a compute-precision descriptor). See https://docs.nvidia.com/cuda/cutensor/index.html. - PFN_
cutensor Create Contraction - Function-pointer type for
cutensorCreateContraction(build an operation descriptor for a tensor contraction). See https://docs.nvidia.com/cuda/cutensor/index.html. - PFN_
cutensor Create Contraction Trinary - Function-pointer type for
cutensorCreateContractionTrinary(build an operation descriptor for a three-tensor contraction). See https://docs.nvidia.com/cuda/cutensor/index.html. - PFN_
cutensor Create Elementwise Binary - Function-pointer type for
cutensorCreateElementwiseBinary(build an operation descriptor for an element-wise binary op). See https://docs.nvidia.com/cuda/cutensor/index.html. - PFN_
cutensor Create Elementwise Trinary - Function-pointer type for
cutensorCreateElementwiseTrinary(build an operation descriptor for an element-wise trinary op). See https://docs.nvidia.com/cuda/cutensor/index.html. - PFN_
cutensor Create Permutation - Function-pointer type for
cutensorCreatePermutation(build an operation descriptor for a tensor permutation). See https://docs.nvidia.com/cuda/cutensor/index.html. - PFN_
cutensor Create Plan - Function-pointer type for
cutensorCreatePlan(build an execution plan from an operation descriptor). See https://docs.nvidia.com/cuda/cutensor/index.html. - PFN_
cutensor Create Plan Preference - Function-pointer type for
cutensorCreatePlanPreference(create a plan-preference object). See https://docs.nvidia.com/cuda/cutensor/index.html. - PFN_
cutensor Create Reduction - Function-pointer type for
cutensorCreateReduction(build an operation descriptor for a tensor reduction). See https://docs.nvidia.com/cuda/cutensor/index.html. - PFN_
cutensor Create Tensor Descriptor - Function-pointer type for
cutensorCreateTensorDescriptor(create a tensor descriptor). See https://docs.nvidia.com/cuda/cutensor/index.html. - PFN_
cutensor Destroy - Function-pointer type for
cutensorDestroy(destroy cuTENSOR library handle). See https://docs.nvidia.com/cuda/cutensor/index.html. - PFN_
cutensor Destroy Block Sparse Tensor Descriptor - Function-pointer type for
cutensorDestroyBlockSparseTensorDescriptor(destroy a block-sparse tensor descriptor). See https://docs.nvidia.com/cuda/cutensor/index.html. - PFN_
cutensor Destroy Compute Descriptor - Function-pointer type for
cutensorDestroyComputeDescriptor(destroy a compute-precision descriptor). See https://docs.nvidia.com/cuda/cutensor/index.html. - PFN_
cutensor Destroy Operation Descriptor - Function-pointer type for
cutensorDestroyOperationDescriptor(destroy an operation descriptor). See https://docs.nvidia.com/cuda/cutensor/index.html. - PFN_
cutensor Destroy Plan - Function-pointer type for
cutensorDestroyPlan(destroy an execution plan). See https://docs.nvidia.com/cuda/cutensor/index.html. - PFN_
cutensor Destroy Plan Preference - Function-pointer type for
cutensorDestroyPlanPreference(destroy a plan-preference object). See https://docs.nvidia.com/cuda/cutensor/index.html. - PFN_
cutensor Destroy Tensor Descriptor - Function-pointer type for
cutensorDestroyTensorDescriptor(destroy a tensor descriptor). See https://docs.nvidia.com/cuda/cutensor/index.html. - PFN_
cutensor Elementwise Binary Execute - Function-pointer type for
cutensorElementwiseBinaryExecute(execute an element-wise binary plan). See https://docs.nvidia.com/cuda/cutensor/index.html. - PFN_
cutensor Elementwise Trinary Execute - Function-pointer type for
cutensorElementwiseTrinaryExecute(execute an element-wise trinary plan). See https://docs.nvidia.com/cuda/cutensor/index.html. - PFN_
cutensor Estimate Workspace Size - Function-pointer type for
cutensorEstimateWorkspaceSize(estimate workspace bytes required by a plan). See https://docs.nvidia.com/cuda/cutensor/index.html. - PFN_
cutensor GetCudart Version - Function-pointer type for
cutensorGetCudartVersion(query the CUDA Runtime version cuTENSOR was built against). See https://docs.nvidia.com/cuda/cutensor/index.html. - PFN_
cutensor GetError String - Function-pointer type for
cutensorGetErrorString(decode a cutensorStatus_t into a static C string). See https://docs.nvidia.com/cuda/cutensor/index.html. - PFN_
cutensor GetVersion - Function-pointer type for
cutensorGetVersion(query cuTENSOR library version). See https://docs.nvidia.com/cuda/cutensor/index.html. - PFN_
cutensor Handle Read Cache From File - Function-pointer type for
cutensorHandleReadCacheFromFile(read a plan/kernel cache from a file). See https://docs.nvidia.com/cuda/cutensor/index.html. - PFN_
cutensor Handle Resize Plan Cache - Function-pointer type for
cutensorHandleResizePlanCache(resize a handle’s plan cache). See https://docs.nvidia.com/cuda/cutensor/index.html. - PFN_
cutensor Handle Write Cache ToFile - Function-pointer type for
cutensorHandleWriteCacheToFile(write a plan/kernel cache to a file). See https://docs.nvidia.com/cuda/cutensor/index.html. - PFN_
cutensor Logger Force Disable - Function-pointer type for
cutensorLoggerForceDisable(force-disable the logger). See https://docs.nvidia.com/cuda/cutensor/index.html. - PFN_
cutensor Logger Open File - Function-pointer type for
cutensorLoggerOpenFile(open a logger output file by path). See https://docs.nvidia.com/cuda/cutensor/index.html. - PFN_
cutensor Logger SetCallback - Function-pointer type for
cutensorLoggerSetCallback(register a logger callback). See https://docs.nvidia.com/cuda/cutensor/index.html. - PFN_
cutensor Logger SetFile - Function-pointer type for
cutensorLoggerSetFile(redirect logger output to an open FILE*). See https://docs.nvidia.com/cuda/cutensor/index.html. - PFN_
cutensor Logger SetLevel - Function-pointer type for
cutensorLoggerSetLevel(set logger verbosity level). See https://docs.nvidia.com/cuda/cutensor/index.html. - PFN_
cutensor Logger SetMask - Function-pointer type for
cutensorLoggerSetMask(set logger category mask). See https://docs.nvidia.com/cuda/cutensor/index.html. - PFN_
cutensor Operation Descriptor GetAttribute - Function-pointer type for
cutensorOperationDescriptorGetAttribute(get an attribute on an operation descriptor). See https://docs.nvidia.com/cuda/cutensor/index.html. - PFN_
cutensor Operation Descriptor SetAttribute - Function-pointer type for
cutensorOperationDescriptorSetAttribute(set an attribute on an operation descriptor). See https://docs.nvidia.com/cuda/cutensor/index.html. - PFN_
cutensor Operation Estimate Runtime - Function-pointer type for
cutensorOperationEstimateRuntime(estimate runtime in milliseconds for a planned operation). See https://docs.nvidia.com/cuda/cutensor/index.html. - PFN_
cutensor Operation NumAlgos - Function-pointer type for
cutensorOperationNumAlgos(query the number of algorithms available for an operation). See https://docs.nvidia.com/cuda/cutensor/index.html. - PFN_
cutensor Permute - Function-pointer type for
cutensorPermute(execute tensor permutation). See https://docs.nvidia.com/cuda/cutensor/index.html. - PFN_
cutensor Plan GetAttribute - Function-pointer type for
cutensorPlanGetAttribute(get an attribute on a plan). See https://docs.nvidia.com/cuda/cutensor/index.html. - PFN_
cutensor Plan Preference GetAttribute - Function-pointer type for
cutensorPlanPreferenceGetAttribute(get an attribute on a plan-preference object). See https://docs.nvidia.com/cuda/cutensor/index.html. - PFN_
cutensor Plan Preference SetAttribute - Function-pointer type for
cutensorPlanPreferenceSetAttribute(set an attribute on a plan-preference object). See https://docs.nvidia.com/cuda/cutensor/index.html. - PFN_
cutensor Reduce - Function-pointer type for
cutensorReduce(execute tensor reduction). See https://docs.nvidia.com/cuda/cutensor/index.html. - PFN_
cutensor Tensor Descriptor GetAttribute - Function-pointer type for
cutensorTensorDescriptorGetAttribute(get an attribute on a tensor descriptor). See https://docs.nvidia.com/cuda/cutensor/index.html. - PFN_
cutensor Tensor Descriptor SetAttribute - Function-pointer type for
cutensorTensorDescriptorSetAttribute(set an attribute on a tensor descriptor). See https://docs.nvidia.com/cuda/cutensor/index.html. - cutensor
Block Sparse Tensor Descriptor_ t - Opaque block-sparse tensor descriptor.
- cutensor
Compute Descriptor_ t cutensorComputeDescriptor_t— the compute-precision descriptor used on modern cuTENSOR (v2+). Opaque pointer.- cutensor
Handle_ t - Opaque cuTENSOR handle.
- cutensor
Operation Descriptor_ t - Opaque contraction-plan descriptor.
- cutensor
Plan Preference_ t - Opaque plan preference handle.
- cutensor
Plan_ t - Opaque plan (built from an operation descriptor + preference).
- cutensor
Tensor Descriptor_ t - Opaque tensor descriptor.