Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore/ccsrc/runtime/device/
Dkernel_runtime.cc1182 const std::shared_ptr<MemScheduler> &mem_scheduler) { in GenAddrCleanLaunchArgs() argument
1201 if (mem_scheduler != nullptr) { in GenAddrCleanLaunchArgs()
1202 GetOrMallocAddress(mem_scheduler, device_address, input); in GenAddrCleanLaunchArgs()
1223 if (mem_scheduler != nullptr) { in GenAddrCleanLaunchArgs()
1224 GetOrMallocAddress(mem_scheduler, device_address, workspace); in GenAddrCleanLaunchArgs()
1286 void KernelRuntime::GetOrMallocAddress(const std::shared_ptr<MemScheduler> &mem_scheduler, in GetOrMallocAddress() argument
1291 kernel_addr->addr = mem_scheduler->GetOrMalloc(device_address, device_address->size_); in GetOrMallocAddress()
1292 if (mem_scheduler->IsHighPriorityMem(device_address)) { in GetOrMallocAddress()
1298 void KernelRuntime::AssignKernelAddress(const std::shared_ptr<MemScheduler> &mem_scheduler, const A… in AssignKernelAddress() argument
1308 return GenAddrCleanLaunchArgs(cnode, kernel_inputs, mem_scheduler); in AssignKernelAddress()
[all …]
Dkernel_runtime.h147 const std::shared_ptr<MemScheduler> &mem_scheduler, bool mock = false);
149 …void AssignKernelAddress(const std::shared_ptr<MemScheduler> &mem_scheduler, const AnfNodePtr &ker…
152 static void GetOrMallocAddress(const std::shared_ptr<MemScheduler> &mem_scheduler,
154 …void InitGraphInputTensors(const std::shared_ptr<MemScheduler> &mem_scheduler, const session::Kern…
155 …void SyncNodeOutputTensors(const std::shared_ptr<MemScheduler> &mem_scheduler, const session::Kern…
/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/
Dascend_kernel_runtime.cc174 auto mem_scheduler = mem_scheduler_manager_.GetMemScheduler(graph_id); in ClearGraphRuntimeResource() local
175 if (mem_scheduler != nullptr) { in ClearGraphRuntimeResource()
176 mem_scheduler->Clear(); in ClearGraphRuntimeResource()