The nvmath.bindings module is experimental and potentially subject to future changes.

cuTENSOR (nvmath.bindings.cutensor)

For detailed documentation on the original C APIs, refer to the cuTENSOR documentation.

Enums and constants

ComputeDesc()	See cutensorComputeDescriptor_t.
cuTENSORError(status)
Operator(*values)	See cutensorOperator_t.
Status(*values)	See cutensorStatus_t.
Algo(*values)	See cutensorAlgo_t.
WorksizePreference(*values)	See cutensorWorksizePreference_t.
OperationDescriptorAttribute(*values)	See cutensorOperationDescriptorAttribute_t.
PlanPreferenceAttribute(*values)	See cutensorPlanPreferenceAttribute_t.
AutotuneMode(*values)	See cutensorAutotuneMode_t.
JitMode(*values)	See cutensorJitMode_t.
CacheMode(*values)	See cutensorCacheMode_t.
PlanAttribute(*values)	See cutensorPlanAttribute_t.

Functions

create()	See cutensorCreate.
destroy(intptr_t handle)	See cutensorDestroy.
handle_resize_plan_cache(intptr_t handle, ...)	See cutensorHandleResizePlanCache.
handle_write_plan_cache_to_file(...)	See cutensorHandleWritePlanCacheToFile.
handle_read_plan_cache_from_file(...)	See cutensorHandleReadPlanCacheFromFile.
write_kernel_cache_to_file(intptr_t handle, ...)	See cutensorWriteKernelCacheToFile.
read_kernel_cache_from_file(intptr_t handle, ...)	See cutensorReadKernelCacheFromFile.
create_tensor_descriptor(intptr_t handle, ...)	See cutensorCreateTensorDescriptor.
destroy_tensor_descriptor(intptr_t desc)	See cutensorDestroyTensorDescriptor.
create_elementwise_trinary(intptr_t handle, ...)	See cutensorCreateElementwiseTrinary.
elementwise_trinary_execute(intptr_t handle, ...)	See cutensorElementwiseTrinaryExecute.
create_elementwise_binary(intptr_t handle, ...)	See cutensorCreateElementwiseBinary.
elementwise_binary_execute(intptr_t handle, ...)	See cutensorElementwiseBinaryExecute.
create_permutation(intptr_t handle, ...)	See cutensorCreatePermutation.
permute(intptr_t handle, intptr_t plan, ...)	See cutensorPermute.
create_contraction(intptr_t handle, ...)	See cutensorCreateContraction.
destroy_operation_descriptor(intptr_t desc)	See cutensorDestroyOperationDescriptor.
get_operation_descriptor_attribute_dtype(...)	Get the Python data type of the corresponding OperationDescriptorAttribute attribute.
operation_descriptor_set_attribute(...)	See cutensorOperationDescriptorSetAttribute.
operation_descriptor_get_attribute(...)	See cutensorOperationDescriptorGetAttribute.
create_plan_preference(intptr_t handle, ...)	See cutensorCreatePlanPreference.
destroy_plan_preference(intptr_t pref)	See cutensorDestroyPlanPreference.
get_plan_preference_attribute_dtype(int attr)	Get the Python data type of the corresponding PlanPreferenceAttribute attribute.
plan_preference_set_attribute(...)	See cutensorPlanPreferenceSetAttribute.
plan_preference_get_attribute(...)	See cutensorPlanPreferenceGetAttribute.
get_plan_attribute_dtype(int attr)	Get the Python data type of the corresponding PlanAttribute attribute.
plan_get_attribute(intptr_t handle, ...)	See cutensorPlanGetAttribute.
estimate_workspace_size(intptr_t handle, ...)	See cutensorEstimateWorkspaceSize.
create_plan(intptr_t handle, intptr_t desc, ...)	See cutensorCreatePlan.
destroy_plan(intptr_t plan)	See cutensorDestroyPlan.
contract(intptr_t handle, intptr_t plan, ...)	See cutensorContract.
create_reduction(intptr_t handle, ...)	See cutensorCreateReduction.
reduce(intptr_t handle, intptr_t plan, ...)	See cutensorReduce.
create_contraction_trinary(intptr_t handle, ...)	See cutensorCreateContractionTrinary.
contract_trinary(intptr_t handle, ...)	See cutensorContractTrinary.
create_block_sparse_tensor_descriptor(...)	See cutensorCreateBlockSparseTensorDescriptor.
destroy_block_sparse_tensor_descriptor(...)	See cutensorDestroyBlockSparseTensorDescriptor.
create_block_sparse_contraction(...)	See cutensorCreateBlockSparseContraction.
block_sparse_contract(intptr_t handle, ...)	See cutensorBlockSparseContract.
get_error_string(int error)	See cutensorGetErrorString.
get_version()	See cutensorGetVersion.
get_cudart_version()	See cutensorGetCudartVersion.
logger_set_file(intptr_t file)	See cutensorLoggerSetFile.
logger_open_file(log_file)	See cutensorLoggerOpenFile.
logger_set_level(int32_t level)	See cutensorLoggerSetLevel.
logger_set_mask(int32_t mask)	See cutensorLoggerSetMask.
logger_force_disable()	See cutensorLoggerForceDisable.