cuquantum.cutensornet.contraction_optimizer_config_set_attribute

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

Set the optimizer config attribute.

Parameters
  • handle (intptr_t) – The library handle.

  • config (intptr_t) – The optimizer config handle.

  • attr (ContractionOptimizerConfigAttribute) – 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_config_get_attribute_dtype().