Searched defs:kernel_pack (Results 1 – 10 of 10) sorted by relevance
36 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()
39 void AkgGpuKernelBuilder::AkgSetKernelMod(const KernelPackPtr &kernel_pack, in AkgSetKernelMod()
37 void AkgAscendKernelBuilder::AkgSetKernelMod(const KernelPackPtr &kernel_pack, in AkgSetKernelMod()
38 AkgKernelMod::AkgKernelMod(const KernelPackPtr &kernel_pack) : kernel_pack_(kernel_pack) {} in AkgKernelMod()
31 explicit TbeKernelMod(KernelPackPtr kernel_pack) : kernel_pack_(std::move(kernel_pack)) {} in TbeKernelMod()
177 auto kernel_pack = TbeUtils::InsertCache(json_name, processor); in TaskFinishProcess() local259 auto kernel_pack = TbeUtils::SearchCache(task_info.json_name); in GenSameFusionOpKernelMod() local
277 uintptr_t KernelManager::GenFuncStub(const mindspore::kernel::KernelPack &kernel_pack, bool force_r… in GenFuncStub()331 std::string KernelManager::GetStubFuncName(const KernelPackPtr &kernel_pack) { in GetStubFuncName()
679 auto kernel_pack = TbeUtils::SearchCache(json_name); in AscendFusionOpCompile() local
95 auto kernel_pack = TbeUtils::SearchCache(json_name); in KernelFusion() local
221 KernelPackPtr kernel_pack = std::make_shared<KernelPack>(); in SearchCache() local245 KernelPackPtr kernel_pack = std::make_shared<KernelPack>(); in InsertCache() local