Home
last modified time | relevance | path

Searched refs:workspace_lists (Results 1 – 1 of 1) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/runtime/device/
Dkernel_runtime.cc356 auto workspace_lists = kernel_mod->GetWorkspaceSizeList(); in RunOpClearMemory() local
357 for (size_t index = 0; index < workspace_lists.size(); ++index) { in RunOpClearMemory()
490 auto workspace_lists = kernel_mod->GetWorkspaceSizeList(); in RunOpAssignWorkSpaceMemory() local
491 for (size_t i = 0; i < workspace_lists.size(); ++i) { in RunOpAssignWorkSpaceMemory()
492 auto device_address = CreateDeviceAddress(nullptr, workspace_lists[i], "", kTypeUnknown); in RunOpAssignWorkSpaceMemory()
494 auto ret = mem_manager_->MallocMemFromMemPool(device_address, workspace_lists[i]); in RunOpAssignWorkSpaceMemory()
496 …LOG(EXCEPTION) << "Device memory isn't enough and alloc failed, alloc size:" << workspace_lists[i]; in RunOpAssignWorkSpaceMemory()