operation_descriptor_set_attribute#

nvmath.bindings.cutensor.operation_descriptor_set_attribute( intptr_t handle, intptr_t desc, int attr, intptr_t buf, size_t size_in_bytes, )[source]#

Set attribute of a cutensorOperationDescriptor_t object.

Parameters:

handle (intptr_t) – Opaque handle holding cuTENSOR’s library context.
desc (intptr_t) – Operation descriptor that will be modified.
attr (OperationDescriptorAttribute) – Specifies the attribute that will be set.
buf (intptr_t) – This buffer (of size size_in_bytes) determines the value to which attr will be set.
size_in_bytes (size_t) – Size of buf (in bytes).

Note

To compute the attribute size, use the itemsize of the corresponding data type, which can be queried using get_operation_descriptor_attribute_dtype().