Struct Cutensor 

pub struct Cutensor {
    pub lib: Library,
    /* private fields */
}

Lazily-resolved cuTENSOR function-pointer table.

Fields

lib: Library

Lib field.

Implementations

impl Cutensor

pub fn compute_desc_32f( &self, ) -> Result<cutensorComputeDescriptor_t, LoaderError>

CUTENSOR_COMPUTE_DESC_32F — 32-bit float compute.

pub fn compute_desc_64f( &self, ) -> Result<cutensorComputeDescriptor_t, LoaderError>

CUTENSOR_COMPUTE_DESC_64F — 64-bit float compute.

pub fn compute_desc_16f( &self, ) -> Result<cutensorComputeDescriptor_t, LoaderError>

CUTENSOR_COMPUTE_DESC_16F — 16-bit float compute.

pub fn compute_desc_16bf( &self, ) -> Result<cutensorComputeDescriptor_t, LoaderError>

CUTENSOR_COMPUTE_DESC_16BF — bfloat16 compute.

pub fn compute_desc_tf32( &self, ) -> Result<cutensorComputeDescriptor_t, LoaderError>

CUTENSOR_COMPUTE_DESC_TF32 — TensorFloat32 compute.

pub fn compute_desc_3xtf32( &self, ) -> Result<cutensorComputeDescriptor_t, LoaderError>

CUTENSOR_COMPUTE_DESC_3XTF32 — 3× TF32 mantissa-extended compute.

pub fn compute_desc_4x16f( &self, ) -> Result<cutensorComputeDescriptor_t, LoaderError>

CUTENSOR_COMPUTE_DESC_4X16F — 4× FP16 mantissa-extended compute.

pub fn compute_desc_8xint8( &self, ) -> Result<cutensorComputeDescriptor_t, LoaderError>

CUTENSOR_COMPUTE_DESC_8XINT8 — 8× INT8 packed compute.

pub fn compute_desc_9x16bf( &self, ) -> Result<cutensorComputeDescriptor_t, LoaderError>

CUTENSOR_COMPUTE_DESC_9X16BF — 9× BF16 mantissa-extended compute.

impl Cutensor

pub fn cutensor_create(&self) -> Result<PFN_cutensorCreate, LoaderError>

Resolve cutensorCreate.

pub fn cutensor_destroy(&self) -> Result<PFN_cutensorDestroy, LoaderError>

Resolve cutensorDestroy.

pub fn cutensor_create_tensor_descriptor( &self, ) -> Result<PFN_cutensorCreateTensorDescriptor, LoaderError>

Resolve cutensorCreateTensorDescriptor.

pub fn cutensor_destroy_tensor_descriptor( &self, ) -> Result<PFN_cutensorDestroyTensorDescriptor, LoaderError>

Resolve cutensorDestroyTensorDescriptor.

pub fn cutensor_create_contraction( &self, ) -> Result<PFN_cutensorCreateContraction, LoaderError>

Resolve cutensorCreateContraction.

pub fn cutensor_destroy_operation_descriptor( &self, ) -> Result<PFN_cutensorDestroyOperationDescriptor, LoaderError>

Resolve cutensorDestroyOperationDescriptor.

pub fn cutensor_create_plan_preference( &self, ) -> Result<PFN_cutensorCreatePlanPreference, LoaderError>

Resolve cutensorCreatePlanPreference.

pub fn cutensor_destroy_plan_preference( &self, ) -> Result<PFN_cutensorDestroyPlanPreference, LoaderError>

Resolve cutensorDestroyPlanPreference.

pub fn cutensor_estimate_workspace_size( &self, ) -> Result<PFN_cutensorEstimateWorkspaceSize, LoaderError>

Resolve cutensorEstimateWorkspaceSize.

pub fn cutensor_create_plan( &self, ) -> Result<PFN_cutensorCreatePlan, LoaderError>

Resolve cutensorCreatePlan.

pub fn cutensor_destroy_plan( &self, ) -> Result<PFN_cutensorDestroyPlan, LoaderError>

Resolve cutensorDestroyPlan.

pub fn cutensor_contract(&self) -> Result<PFN_cutensorContract, LoaderError>

Resolve cutensorContract.

pub fn cutensor_get_version( &self, ) -> Result<PFN_cutensorGetVersion, LoaderError>

Resolve cutensorGetVersion.

pub fn cutensor_get_cudart_version( &self, ) -> Result<PFN_cutensorGetCudartVersion, LoaderError>

Resolve cutensorGetCudartVersion.

pub fn cutensor_get_error_string( &self, ) -> Result<PFN_cutensorGetErrorString, LoaderError>

Resolve cutensorGetErrorString.

pub fn cutensor_create_elementwise_binary( &self, ) -> Result<PFN_cutensorCreateElementwiseBinary, LoaderError>

Resolve cutensorCreateElementwiseBinary.

pub fn cutensor_elementwise_binary_execute( &self, ) -> Result<PFN_cutensorElementwiseBinaryExecute, LoaderError>

Resolve cutensorElementwiseBinaryExecute.

pub fn cutensor_create_elementwise_trinary( &self, ) -> Result<PFN_cutensorCreateElementwiseTrinary, LoaderError>

Resolve cutensorCreateElementwiseTrinary.

pub fn cutensor_elementwise_trinary_execute( &self, ) -> Result<PFN_cutensorElementwiseTrinaryExecute, LoaderError>

Resolve cutensorElementwiseTrinaryExecute.

pub fn cutensor_create_permutation( &self, ) -> Result<PFN_cutensorCreatePermutation, LoaderError>

Resolve cutensorCreatePermutation.

pub fn cutensor_permute(&self) -> Result<PFN_cutensorPermute, LoaderError>

Resolve cutensorPermute.

pub fn cutensor_create_reduction( &self, ) -> Result<PFN_cutensorCreateReduction, LoaderError>

Resolve cutensorCreateReduction.

pub fn cutensor_reduce(&self) -> Result<PFN_cutensorReduce, LoaderError>

Resolve cutensorReduce.

pub fn cutensor_operation_descriptor_get_attribute( &self, ) -> Result<PFN_cutensorOperationDescriptorGetAttribute, LoaderError>

Resolve cutensorOperationDescriptorGetAttribute.

pub fn cutensor_operation_descriptor_set_attribute( &self, ) -> Result<PFN_cutensorOperationDescriptorSetAttribute, LoaderError>

Resolve cutensorOperationDescriptorSetAttribute.

pub fn cutensor_plan_preference_set_attribute( &self, ) -> Result<PFN_cutensorPlanPreferenceSetAttribute, LoaderError>

Resolve cutensorPlanPreferenceSetAttribute.

pub fn cutensor_plan_get_attribute( &self, ) -> Result<PFN_cutensorPlanGetAttribute, LoaderError>

Resolve cutensorPlanGetAttribute.

pub fn cutensor_tensor_descriptor_get_attribute( &self, ) -> Result<PFN_cutensorTensorDescriptorGetAttribute, LoaderError>

Resolve cutensorTensorDescriptorGetAttribute.

pub fn cutensor_handle_resize_plan_cache( &self, ) -> Result<PFN_cutensorHandleResizePlanCache, LoaderError>

Resolve cutensorHandleResizePlanCache.

pub fn cutensor_handle_read_plan_cache_from_file( &self, ) -> Result<PFN_cutensorHandleReadCacheFromFile, LoaderError>

Resolve cutensorHandleReadPlanCacheFromFile.

pub fn cutensor_handle_write_plan_cache_to_file( &self, ) -> Result<PFN_cutensorHandleWriteCacheToFile, LoaderError>

Resolve cutensorHandleWritePlanCacheToFile.

pub fn cutensor_read_kernel_cache_from_file( &self, ) -> Result<PFN_cutensorHandleReadCacheFromFile, LoaderError>

Resolve cutensorReadKernelCacheFromFile.

pub fn cutensor_write_kernel_cache_to_file( &self, ) -> Result<PFN_cutensorHandleWriteCacheToFile, LoaderError>

Resolve cutensorWriteKernelCacheToFile.

pub fn cutensor_create_contraction_trinary( &self, ) -> Result<PFN_cutensorCreateContractionTrinary, LoaderError>

Resolve cutensorCreateContractionTrinary.

pub fn cutensor_contract_trinary( &self, ) -> Result<PFN_cutensorContractTrinary, LoaderError>

Resolve cutensorContractTrinary.

pub fn cutensor_create_compute_descriptor( &self, ) -> Result<PFN_cutensorCreateComputeDescriptor, LoaderError>

Resolve cutensorCreateComputeDescriptor.

pub fn cutensor_destroy_compute_descriptor( &self, ) -> Result<PFN_cutensorDestroyComputeDescriptor, LoaderError>

Resolve cutensorDestroyComputeDescriptor.

pub fn cutensor_compute_descriptor_get_attribute( &self, ) -> Result<PFN_cutensorComputeDescriptorGetAttribute, LoaderError>

Resolve cutensorComputeDescriptorGetAttribute.

pub fn cutensor_compute_descriptor_set_attribute( &self, ) -> Result<PFN_cutensorComputeDescriptorSetAttribute, LoaderError>

Resolve cutensorComputeDescriptorSetAttribute.

pub fn cutensor_tensor_descriptor_set_attribute( &self, ) -> Result<PFN_cutensorTensorDescriptorSetAttribute, LoaderError>

Resolve cutensorTensorDescriptorSetAttribute.

pub fn cutensor_plan_preference_get_attribute( &self, ) -> Result<PFN_cutensorPlanPreferenceGetAttribute, LoaderError>

Resolve cutensorPlanPreferenceGetAttribute.

pub fn cutensor_operation_estimate_runtime( &self, ) -> Result<PFN_cutensorOperationEstimateRuntime, LoaderError>

Resolve cutensorOperationEstimateRuntime.

pub fn cutensor_operation_num_algos( &self, ) -> Result<PFN_cutensorOperationNumAlgos, LoaderError>

Resolve cutensorOperationNumAlgos.

pub fn cutensor_logger_set_level( &self, ) -> Result<PFN_cutensorLoggerSetLevel, LoaderError>

Resolve cutensorLoggerSetLevel.

pub fn cutensor_logger_set_mask( &self, ) -> Result<PFN_cutensorLoggerSetMask, LoaderError>

Resolve cutensorLoggerSetMask.

pub fn cutensor_logger_open_file( &self, ) -> Result<PFN_cutensorLoggerOpenFile, LoaderError>

Resolve cutensorLoggerOpenFile.

pub fn cutensor_logger_set_file( &self, ) -> Result<PFN_cutensorLoggerSetFile, LoaderError>

Resolve cutensorLoggerSetFile.

pub fn cutensor_logger_set_callback( &self, ) -> Result<PFN_cutensorLoggerSetCallback, LoaderError>

Resolve cutensorLoggerSetCallback.

pub fn cutensor_logger_force_disable( &self, ) -> Result<PFN_cutensorLoggerForceDisable, LoaderError>

Resolve cutensorLoggerForceDisable.

pub fn cutensor_create_block_sparse_tensor_descriptor( &self, ) -> Result<PFN_cutensorCreateBlockSparseTensorDescriptor, LoaderError>

Resolve cutensorCreateBlockSparseTensorDescriptor.

pub fn cutensor_destroy_block_sparse_tensor_descriptor( &self, ) -> Result<PFN_cutensorDestroyBlockSparseTensorDescriptor, LoaderError>

Resolve cutensorDestroyBlockSparseTensorDescriptor.

pub fn cutensor_create_block_sparse_contraction( &self, ) -> Result<PFN_cutensorCreateBlockSparseContraction, LoaderError>

Resolve cutensorCreateBlockSparseContraction.

pub fn cutensor_block_sparse_contract( &self, ) -> Result<PFN_cutensorBlockSparseContract, LoaderError>

Resolve cutensorBlockSparseContract.

Trait Implementations

impl Debug for Cutensor

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.