cuquantum.cutensornet.workspace_set_memory¶
- cuquantum.cutensornet.workspace_set_memory(intptr_t handle, intptr_t work_desc, int mem_space, int work_kind, intptr_t memory_ptr, int64_t memory_size)[source]¶
设置用户提供的工作空间的内存地址和工作空间大小。
- 参数
handle (intptr_t) – 持有 cuTensorNet 库上下文的不透明句柄。
work_desc (intptr_t) – 描述工作空间的不透明结构。
mem_space (Memspace) – 分配工作空间的内存空间。
work_kind (WorkspaceKind) – 工作空间的种类。
memory_ptr (intptr_t) – 工作空间内存指针，可以为空。
memory_size (int64_t) – 工作空间大小。