Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/
Dascend_kernel_runtime.cc116 std::vector<rtExceptionInfo> AscendKernelRuntime::task_fail_infoes_ = {};
118 std::map<std::string, uint32_t> AscendKernelRuntime::overflow_tasks_;
119 AscendKernelRuntime::~AscendKernelRuntime() { in ~AscendKernelRuntime()
125 void AscendKernelRuntime::SetContext() { in SetContext()
139 void AscendKernelRuntime::SetCurrentContext() { in SetCurrentContext()
149 void AscendKernelRuntime::ClearGraphModelMap() { in ClearGraphModelMap()
172 void AscendKernelRuntime::ClearGraphRuntimeResource(uint32_t graph_id) { in ClearGraphRuntimeResource()
212 void *AscendKernelRuntime::GetModelStream(uint32_t graph_id) const { in GetModelStream()
216 void AscendKernelRuntime::ClearGlobalIdleMem() { in ClearGlobalIdleMem()
222 bool AscendKernelRuntime::NeedDestroyHccl() { in NeedDestroyHccl()
[all …]
Dascend_kernel_runtime.h39 class AscendKernelRuntime : public KernelRuntime {
41 AscendKernelRuntime() = default;
42 ~AscendKernelRuntime() override;
125 MS_REG_KERNEL_RUNTIME(kAscendDevice, AscendKernelRuntime);
/third_party/mindspore/mindspore/ccsrc/runtime/device/
Ddevice_address.h39 class AscendKernelRuntime; variable
125 friend class mindspore::device::ascend::AscendKernelRuntime;