Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore/ccsrc/backend/session/
Dkernel_build_client.h194 class AscendKernelBuildClient : public KernelBuildClient {
225 static AscendKernelBuildClient &Instance() { in Instance()
226 static AscendKernelBuildClient instance; in Instance()
247 AscendKernelBuildClient(const AscendKernelBuildClient &) = delete;
248 AscendKernelBuildClient &operator=(const AscendKernelBuildClient &) = delete;
250 AscendKernelBuildClient(AscendKernelBuildClient &&) = delete;
251 AscendKernelBuildClient &operator=(AscendKernelBuildClient &&) = delete;
255 AscendKernelBuildClient() { Open(); } in AscendKernelBuildClient() function
256 ~AscendKernelBuildClient() override { Close(); } in ~AscendKernelBuildClient()
Dkernel_build_client.cc93 void AscendKernelBuildClient::TbePre(const std::string &mode) { in TbePre()
110 int AscendKernelBuildClient::TbeStart(const std::string &json, const std::string &mode) { in TbeStart()
131 std::string AscendKernelBuildClient::TbeSendJob(const std::string &json) { in TbeSendJob()
146 bool AscendKernelBuildClient::TbeWait(int *task_id, std::string *task_result, std::string *pre_buil… in TbeWait()
162 void AscendKernelBuildClient::TbeReset() { in TbeReset()
171 std::string AscendKernelBuildClient::SelectFormat(const std::string &json) { in SelectFormat()
187 bool AscendKernelBuildClient::CheckSupported(const std::string &json) { in CheckSupported()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/akg/ascend/
Dakg_ascend_kernel_build.h34 …kernel::KernelBuildClient *GetClient() override { return &(kernel::AscendKernelBuildClient::Instan… in GetClient()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/
Dtbe_kernel_parallel_build.cc301 return AscendKernelBuildClient::Instance().TbeStart(kernel_json.dump(), tune_mode); in StartCompileOp()
305 return AscendKernelBuildClient::Instance().TbeSendJob(kernel_json.dump()); in ProcessTbeJob()
310 return AscendKernelBuildClient::Instance().TbeWait(task_id, task_result, pre_build_result); in WaitOne()
Dtbe_kernel_parallel_build.h44 ParallelBuildManager() { AscendKernelBuildClient::Instance().TbeReset(); } in ParallelBuildManager()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/tbe_kernel_select/
Dtbe_kernel_select.cc281 ret = AscendKernelBuildClient::Instance().CheckSupported(kernel_json.dump()); in TbeCheckSupported()
455 res_json_str = AscendKernelBuildClient::Instance().SelectFormat(kernel_json.dump()); in OpSelectFormat()
/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/
Dascend_kernel_runtime.cc109 mindspore::kernel::AscendKernelBuildClient::Instance().Close(); in IntHandler()