TensorDescriptorAttribute

enum cuquantum.bindings.cutensornet.TensorDescriptorAttribute(value)

This enum lists attributes of a cutensornetTensorDescriptor_t that can be queried via cutensornetTensorDescriptorGetAttribute(). Array- valued attributes expect a buffer of length numModes (as returned by cutensornetGetTensorDetails()). For a non-distributed descriptor, distributed-only attributes are still defined: block sizes and nranksPerMode entries are treated as fully local (nranksPerMode[i] == 1, block size 0 meaning a single contiguous partition).

See cutensornetTensorDescriptorAttributes_t.

Member Type:

int

Valid values are as follows:

IS_DISTRIBUTED = <TensorDescriptorAttribute.IS_DISTRIBUTED: 0>

ELEMENT_STRIDES = <TensorDescriptorAttribute.ELEMENT_STRIDES: 1>

BLOCK_SIZES = <TensorDescriptorAttribute.BLOCK_SIZES: 2>

BLOCK_STRIDES = <TensorDescriptorAttribute.BLOCK_STRIDES: 3>

NRANKS_PER_MODE = <TensorDescriptorAttribute.NRANKS_PER_MODE: 4>

LOCAL_DATA_SIZE = <TensorDescriptorAttribute.LOCAL_DATA_SIZE: 5>

LOCAL_EXTENTS = <TensorDescriptorAttribute.LOCAL_EXTENTS: 6>