cuquantum.cutensornet.workspace_get_memory

cuquantum.cutensornet.workspace_get_memory(intptr_t handle, intptr_t work_desc, int mem_space, int work_kind) → tuple[源代码]

检索工作区描述符中托管的工作区的内存地址和工作区大小。

参数

  • handle (intptr_t) – 不透明句柄，用于保存 cuTensorNet 的库上下文。

  • work_desc (intptr_t) – 描述工作区的不透明结构。

  • mem_space (Memspace) – 工作区分配到的内存空间。

  • work_kind (WorkspaceKind) – 工作区的类型。

返回

  • intptr_t: 工作区内存指针。

  • int64_t: 工作区大小。

返回类型

包含以下内容的 2 元组

另请参阅

cutensornetWorkspaceGetMemory()