Python functions

Handle Management API

create()

Create a cuTensorNet handle.

destroy(intptr_t handle)

Destroy a cuTensorNet handle.

Network Descriptor API

create_network_descriptor(intptr_t handle, ...)

Create a tensor network descriptor.

destroy_network_descriptor(intptr_t tn_desc)

Destroy a tensor network descriptor.

Contraction Optimizer API

create_contraction_optimizer_info(...)

Create a contraction optimizer info object.

destroy_contraction_optimizer_info(intptr_t info)

Destroy a contraction optimizer info object.

contraction_optimizer_info_get_attribute_dtype(...)

Get the Python data type of the corresponding optimizer info attribute.

contraction_optimizer_info_get_attribute(...)

Get the optimizer info attribute.

contraction_optimizer_info_set_attribute(...)

Set the optimizer info attribute.

create_contraction_optimizer_config(...)

Create a contraction optimizer config object.

destroy_contraction_optimizer_config(...)

Destroy a contraction optimizer config object.

contraction_optimizer_config_get_attribute_dtype(...)

Get the Python data type of the corresponding optimizer config attribute.

contraction_optimizer_config_get_attribute(...)

Get the optimizer config attribute.

contraction_optimizer_config_set_attribute(...)

Set the optimizer config attribute.

contraction_optimize(intptr_t handle, ...)

Optimize the contraction path, slicing, etc, for the given tensor network.

Contraction Plan API

create_contraction_plan(intptr_t handle, ...)

Create a contraction plan for the given tensor network and the associated path.

destroy_contraction_plan(intptr_t plan)

Destroy a contraction plan.

create_contraction_autotune_preference(...)

Create a handle to hold all autotune parameters.

destroy_contraction_autotune_preference(...)

Destroy the autotue preference handle.

contraction_autotune_preference_get_attribute_dtype(...)

Get the Python data type of the corresponding autotune preference attribute.

contraction_autotune_preference_get_attribute(...)

Get the autotue preference attributes.

contraction_autotune_preference_set_attribute(...)

Set the autotue preference attributes.

contraction_autotune(intptr_t handle, ...)

Autotune the contraction plan to find the best kernels for each pairwise tensor contraction.

Network Contraction API

contraction_get_workspace_size(...)

Compute the required workspace size for contracting the input tensor network.

contraction(intptr_t handle, intptr_t plan, ...)

Perform the contraction of the input tensors.

Versioning API

get_version()

Query the version of the cuTensorNet library.

get_cudart_version()

Query the version of the CUDA runtime.