Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/
Dtbe_utils.h65 static KernelPackPtr SearchCache(const std::string &kernel_name, const bool is_akg = false);
67 static KernelPackPtr InsertCache(const std::string &kernel_name, const std::string &processor,
82 static std::string GetStubFuncName(const KernelPackPtr &kernel_pack);
97 KernelPackPtr GetKernelPack(const std::string &kernel_name, const bool is_akg = false);
103 std::unordered_map<std::string, KernelPackPtr> kernel_pack_map_{};
Dtbe_kernel_mod.h31 explicit TbeKernelMod(KernelPackPtr kernel_pack) : kernel_pack_(std::move(kernel_pack)) {} in TbeKernelMod()
49 KernelPackPtr kernel_pack_;
Dtbe_utils.cc231 KernelPackPtr TbeUtils::SearchCache(const std::string &kernel_name, const bool is_akg) { in SearchCache()
241 KernelPackPtr TbeUtils::InsertCache(const std::string &kernel_name, const std::string &processor, c… in InsertCache()
331 std::string KernelManager::GetStubFuncName(const KernelPackPtr &kernel_pack) { in GetStubFuncName()
500 KernelPackPtr KernelMeta::GetKernelPack(const std::string &kernel_name, const bool is_akg) { in GetKernelPack()
501 KernelPackPtr ret = nullptr; in GetKernelPack()
Dtbe_kernel_parallel_build.h63 const KernelPackPtr &kernel_pack) const;
Dtbe_kernel_parallel_build.cc288 … const mindspore::kernel::KernelPackPtr &kernel_pack) const { in GenKernelMod()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/akg/gpu/
Dakg_gpu_kernel_build.h31 KernelPackPtr AkgSearchCache(const std::string &kernel_name) override;
32 KernelPackPtr AkgInsertCache(const std::string &kernel_name) override;
33 …void AkgSetKernelMod(const KernelPackPtr &kernel_pack, const AkgKernelJsonGenerator &json_generato…
Dakg_gpu_kernel_build.cc31 KernelPackPtr AkgGpuKernelBuilder::AkgSearchCache(const std::string &kernel_name) { in AkgSearchCache()
35 KernelPackPtr AkgGpuKernelBuilder::AkgInsertCache(const std::string &kernel_name) { in AkgInsertCache()
39 void AkgGpuKernelBuilder::AkgSetKernelMod(const KernelPackPtr &kernel_pack, in AkgSetKernelMod()
Dakg_gpu_kernel_mod.h48 …CUresult GetFunction(const KernelPackPtr &kernel_pack, bool force_reload, std::vector<uint32_t> *t…
58 explicit GpuKernelMod(const KernelPackPtr &kernel_pack);
73 KernelPackPtr kernel_pack_;
Dakg_gpu_kernel_mod.cc36 CUresult GpuKernelManager::GetFunction(const KernelPackPtr &kernel_pack, bool force_reload, in GetFunction()
88 GpuKernelMod::GpuKernelMod(const KernelPackPtr &kernel_pack) : kernel_pack_(kernel_pack) {} in GpuKernelMod()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/akg/ascend/
Dakg_ascend_kernel_build.h35 KernelPackPtr AkgSearchCache(const std::string &kernel_name) override;
36 KernelPackPtr AkgInsertCache(const std::string &kernel_name) override;
37 …void AkgSetKernelMod(const KernelPackPtr &kernel_pack, const AkgKernelJsonGenerator &json_generato…
Dakg_ascend_kernel_build.cc29 KernelPackPtr AkgAscendKernelBuilder::AkgSearchCache(const std::string &kernel_name) { in AkgSearchCache()
33 KernelPackPtr AkgAscendKernelBuilder::AkgInsertCache(const std::string &kernel_name) { in AkgInsertCache()
37 void AkgAscendKernelBuilder::AkgSetKernelMod(const KernelPackPtr &kernel_pack, in AkgSetKernelMod()
Dakg_ascend_kernel_mod.h29 explicit AkgKernelMod(const KernelPackPtr &kernel_pack);
44 KernelPackPtr kernel_pack_;
Dakg_ascend_kernel_mod.cc38 AkgKernelMod::AkgKernelMod(const KernelPackPtr &kernel_pack) : kernel_pack_(kernel_pack) {} in AkgKernelMod()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/akg/
Dakg_kernel_build.h42 virtual KernelPackPtr AkgSearchCache(const std::string &kernel_name) = 0;
43 virtual KernelPackPtr AkgInsertCache(const std::string &kernel_name) = 0;
44 …virtual void AkgSetKernelMod(const KernelPackPtr &kernel_pack, const AkgKernelJsonGenerator &json_…
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/
Dcommon_utils.h76 KernelPackPtr SearchCache(const std::string &kernel_name, const std::string &processor);
77 KernelPackPtr InsertCache(const std::string &kernel_name, const std::string &processor);
Dcommon_utils.cc211 KernelPackPtr SearchCache(const std::string &kernel_name, const std::string &processor) { in SearchCache()
221 KernelPackPtr kernel_pack = std::make_shared<KernelPack>(); in SearchCache()
235 KernelPackPtr InsertCache(const std::string &kernel_name, const std::string &processor) { in InsertCache()
245 KernelPackPtr kernel_pack = std::make_shared<KernelPack>(); in InsertCache()
Dkernel.h157 using KernelPackPtr = std::shared_ptr<KernelPack>; variable