Skip to main content

Crate baracuda_cutensor_sys

Crate baracuda_cutensor_sys 

Source
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§

cutensorAlgo
cutensorAlgo_t — algorithm selector for contraction planning.
cutensorDataType
cutensorDataType_t — element type enum.
cutensorJitMode
cutensorJitMode_t — Just-in-time-compile selector (cuTENSOR 2.x).
cutensorOperator
cutensorOperator_t — element-wise op selector.
cutensorWorksizePreference
cutensorWorksizePreference_t.

Structs§

Cutensor
Lazily-resolved cuTENSOR function-pointer table.
cutensorStatus_t
cuTENSOR status code.

Functions§

cutensor
Return the lazily-loaded cuTENSOR library accessor.

Type Aliases§

PFN_cutensorBlockSparseContract
Function-pointer type for cutensorBlockSparseContract (execute block-sparse tensor contraction). See https://docs.nvidia.com/cuda/cutensor/index.html.
PFN_cutensorComputeDescriptorGetAttribute
Function-pointer type for cutensorComputeDescriptorGetAttribute (get an attribute on a compute descriptor). See https://docs.nvidia.com/cuda/cutensor/index.html.
PFN_cutensorComputeDescriptorSetAttribute
Function-pointer type for cutensorComputeDescriptorSetAttribute (set an attribute on a compute descriptor). See https://docs.nvidia.com/cuda/cutensor/index.html.
PFN_cutensorContract
Function-pointer type for cutensorContract (execute tensor contraction). See https://docs.nvidia.com/cuda/cutensor/index.html.
PFN_cutensorContractTrinary
Function-pointer type for cutensorContractTrinary (execute three-tensor contraction). See https://docs.nvidia.com/cuda/cutensor/index.html.
PFN_cutensorCreate
Function-pointer type for cutensorCreate (create cuTENSOR library handle). See https://docs.nvidia.com/cuda/cutensor/index.html.
PFN_cutensorCreateBlockSparseContraction
Function-pointer type for cutensorCreateBlockSparseContraction (build an operation descriptor for a block-sparse contraction). See https://docs.nvidia.com/cuda/cutensor/index.html.
PFN_cutensorCreateBlockSparseTensorDescriptor
Function-pointer type for cutensorCreateBlockSparseTensorDescriptor (create a block-sparse tensor descriptor). See https://docs.nvidia.com/cuda/cutensor/index.html.
PFN_cutensorCreateComputeDescriptor
Function-pointer type for cutensorCreateComputeDescriptor (create a compute-precision descriptor). See https://docs.nvidia.com/cuda/cutensor/index.html.
PFN_cutensorCreateContraction
Function-pointer type for cutensorCreateContraction (build an operation descriptor for a tensor contraction). See https://docs.nvidia.com/cuda/cutensor/index.html.
PFN_cutensorCreateContractionTrinary
Function-pointer type for cutensorCreateContractionTrinary (build an operation descriptor for a three-tensor contraction). See https://docs.nvidia.com/cuda/cutensor/index.html.
PFN_cutensorCreateElementwiseBinary
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_cutensorCreateElementwiseTrinary
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_cutensorCreatePermutation
Function-pointer type for cutensorCreatePermutation (build an operation descriptor for a tensor permutation). See https://docs.nvidia.com/cuda/cutensor/index.html.
PFN_cutensorCreatePlan
Function-pointer type for cutensorCreatePlan (build an execution plan from an operation descriptor). See https://docs.nvidia.com/cuda/cutensor/index.html.
PFN_cutensorCreatePlanPreference
Function-pointer type for cutensorCreatePlanPreference (create a plan-preference object). See https://docs.nvidia.com/cuda/cutensor/index.html.
PFN_cutensorCreateReduction
Function-pointer type for cutensorCreateReduction (build an operation descriptor for a tensor reduction). See https://docs.nvidia.com/cuda/cutensor/index.html.
PFN_cutensorCreateTensorDescriptor
Function-pointer type for cutensorCreateTensorDescriptor (create a tensor descriptor). See https://docs.nvidia.com/cuda/cutensor/index.html.
PFN_cutensorDestroy
Function-pointer type for cutensorDestroy (destroy cuTENSOR library handle). See https://docs.nvidia.com/cuda/cutensor/index.html.
PFN_cutensorDestroyBlockSparseTensorDescriptor
Function-pointer type for cutensorDestroyBlockSparseTensorDescriptor (destroy a block-sparse tensor descriptor). See https://docs.nvidia.com/cuda/cutensor/index.html.
PFN_cutensorDestroyComputeDescriptor
Function-pointer type for cutensorDestroyComputeDescriptor (destroy a compute-precision descriptor). See https://docs.nvidia.com/cuda/cutensor/index.html.
PFN_cutensorDestroyOperationDescriptor
Function-pointer type for cutensorDestroyOperationDescriptor (destroy an operation descriptor). See https://docs.nvidia.com/cuda/cutensor/index.html.
PFN_cutensorDestroyPlan
Function-pointer type for cutensorDestroyPlan (destroy an execution plan). See https://docs.nvidia.com/cuda/cutensor/index.html.
PFN_cutensorDestroyPlanPreference
Function-pointer type for cutensorDestroyPlanPreference (destroy a plan-preference object). See https://docs.nvidia.com/cuda/cutensor/index.html.
PFN_cutensorDestroyTensorDescriptor
Function-pointer type for cutensorDestroyTensorDescriptor (destroy a tensor descriptor). See https://docs.nvidia.com/cuda/cutensor/index.html.
PFN_cutensorElementwiseBinaryExecute
Function-pointer type for cutensorElementwiseBinaryExecute (execute an element-wise binary plan). See https://docs.nvidia.com/cuda/cutensor/index.html.
PFN_cutensorElementwiseTrinaryExecute
Function-pointer type for cutensorElementwiseTrinaryExecute (execute an element-wise trinary plan). See https://docs.nvidia.com/cuda/cutensor/index.html.
PFN_cutensorEstimateWorkspaceSize
Function-pointer type for cutensorEstimateWorkspaceSize (estimate workspace bytes required by a plan). See https://docs.nvidia.com/cuda/cutensor/index.html.
PFN_cutensorGetCudartVersion
Function-pointer type for cutensorGetCudartVersion (query the CUDA Runtime version cuTENSOR was built against). See https://docs.nvidia.com/cuda/cutensor/index.html.
PFN_cutensorGetErrorString
Function-pointer type for cutensorGetErrorString (decode a cutensorStatus_t into a static C string). See https://docs.nvidia.com/cuda/cutensor/index.html.
PFN_cutensorGetVersion
Function-pointer type for cutensorGetVersion (query cuTENSOR library version). See https://docs.nvidia.com/cuda/cutensor/index.html.
PFN_cutensorHandleReadCacheFromFile
Function-pointer type for cutensorHandleReadCacheFromFile (read a plan/kernel cache from a file). See https://docs.nvidia.com/cuda/cutensor/index.html.
PFN_cutensorHandleResizePlanCache
Function-pointer type for cutensorHandleResizePlanCache (resize a handle’s plan cache). See https://docs.nvidia.com/cuda/cutensor/index.html.
PFN_cutensorHandleWriteCacheToFile
Function-pointer type for cutensorHandleWriteCacheToFile (write a plan/kernel cache to a file). See https://docs.nvidia.com/cuda/cutensor/index.html.
PFN_cutensorLoggerForceDisable
Function-pointer type for cutensorLoggerForceDisable (force-disable the logger). See https://docs.nvidia.com/cuda/cutensor/index.html.
PFN_cutensorLoggerOpenFile
Function-pointer type for cutensorLoggerOpenFile (open a logger output file by path). See https://docs.nvidia.com/cuda/cutensor/index.html.
PFN_cutensorLoggerSetCallback
Function-pointer type for cutensorLoggerSetCallback (register a logger callback). See https://docs.nvidia.com/cuda/cutensor/index.html.
PFN_cutensorLoggerSetFile
Function-pointer type for cutensorLoggerSetFile (redirect logger output to an open FILE*). See https://docs.nvidia.com/cuda/cutensor/index.html.
PFN_cutensorLoggerSetLevel
Function-pointer type for cutensorLoggerSetLevel (set logger verbosity level). See https://docs.nvidia.com/cuda/cutensor/index.html.
PFN_cutensorLoggerSetMask
Function-pointer type for cutensorLoggerSetMask (set logger category mask). See https://docs.nvidia.com/cuda/cutensor/index.html.
PFN_cutensorOperationDescriptorGetAttribute
Function-pointer type for cutensorOperationDescriptorGetAttribute (get an attribute on an operation descriptor). See https://docs.nvidia.com/cuda/cutensor/index.html.
PFN_cutensorOperationDescriptorSetAttribute
Function-pointer type for cutensorOperationDescriptorSetAttribute (set an attribute on an operation descriptor). See https://docs.nvidia.com/cuda/cutensor/index.html.
PFN_cutensorOperationEstimateRuntime
Function-pointer type for cutensorOperationEstimateRuntime (estimate runtime in milliseconds for a planned operation). See https://docs.nvidia.com/cuda/cutensor/index.html.
PFN_cutensorOperationNumAlgos
Function-pointer type for cutensorOperationNumAlgos (query the number of algorithms available for an operation). See https://docs.nvidia.com/cuda/cutensor/index.html.
PFN_cutensorPermute
Function-pointer type for cutensorPermute (execute tensor permutation). See https://docs.nvidia.com/cuda/cutensor/index.html.
PFN_cutensorPlanGetAttribute
Function-pointer type for cutensorPlanGetAttribute (get an attribute on a plan). See https://docs.nvidia.com/cuda/cutensor/index.html.
PFN_cutensorPlanPreferenceGetAttribute
Function-pointer type for cutensorPlanPreferenceGetAttribute (get an attribute on a plan-preference object). See https://docs.nvidia.com/cuda/cutensor/index.html.
PFN_cutensorPlanPreferenceSetAttribute
Function-pointer type for cutensorPlanPreferenceSetAttribute (set an attribute on a plan-preference object). See https://docs.nvidia.com/cuda/cutensor/index.html.
PFN_cutensorReduce
Function-pointer type for cutensorReduce (execute tensor reduction). See https://docs.nvidia.com/cuda/cutensor/index.html.
PFN_cutensorTensorDescriptorGetAttribute
Function-pointer type for cutensorTensorDescriptorGetAttribute (get an attribute on a tensor descriptor). See https://docs.nvidia.com/cuda/cutensor/index.html.
PFN_cutensorTensorDescriptorSetAttribute
Function-pointer type for cutensorTensorDescriptorSetAttribute (set an attribute on a tensor descriptor). See https://docs.nvidia.com/cuda/cutensor/index.html.
cutensorBlockSparseTensorDescriptor_t
Opaque block-sparse tensor descriptor.
cutensorComputeDescriptor_t
cutensorComputeDescriptor_t — the compute-precision descriptor used on modern cuTENSOR (v2+). Opaque pointer.
cutensorHandle_t
Opaque cuTENSOR handle.
cutensorOperationDescriptor_t
Opaque contraction-plan descriptor.
cutensorPlanPreference_t
Opaque plan preference handle.
cutensorPlan_t
Opaque plan (built from an operation descriptor + preference).
cutensorTensorDescriptor_t
Opaque tensor descriptor.