cuquantum.cutensornet.contraction_autotune_preference_set_attribute

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

设置 autotune_preference 的属性。

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

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

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

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

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

注意

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