cuquantum.cutensornet.contraction_optimizer_info_set_attribute

cuquantum.cutensornet.contraction_optimizer_info_set_attribute(intptr_t handle, intptr_t info, int attr, intptr_t buf, size_t size)[source]

Set the optimizer info attribute.

Parameters
  • handle (intptr_t) – The library handle.

  • info (intptr_t) – The optimizer info handle.

  • attr (ContractionOptimizerInfoAttribute) – The attribute to set.

  • buf (intptr_t) – The pointer address (as Python int) to the attribute data.

  • size (size_t) – The size of buf (in bytes).

Note

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