cuquantum.cutensornet.contraction_optimizer_config_set_attribute

cuquantum.cutensornet.contraction_optimizer_config_set_attribute(intptr_t handle, intptr_t optimizer_config, int attr, intptr_t buf, size_t size_in_bytes)[源代码]

设置 optimizer_config 的属性。

参数
  • handle (intptr_t) – 持有 cuTensorNet 库上下文的不透明句柄。

  • optimizer_config (intptr_t) – 要访问的不透明结构。

  • attr (ContractionOptimizerConfigAttribute) – 指定请求的属性。

  • buf (intptr_t) – 此缓冲区(大小为 size_in_bytes)确定 attr 将被设置的值。

  • size_in_bytes (size_t) – buf 的大小(以字节为单位)。

注意

要计算属性大小,请使用相应数据类型的 itemsize,可以使用 get_contraction_optimizer_config_attribute_dtype() 查询。