| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/akg/gpu/ |
| D | akg_gpu_kernel_mod.cc | 36 CUresult GpuKernelManager::GetFunction(const KernelPackPtr &kernel_pack, bool force_reload, in GetFunction() argument
38 if (kernel_pack->GetJson() == nullptr || kernel_pack->GetJson()->contents == nullptr || in GetFunction()
39 kernel_pack->GetKernel() == nullptr || kernel_pack->GetKernel()->contents == nullptr) { in GetFunction()
43 auto js = nlohmann::json::parse(kernel_pack->GetJson()->contents, in GetFunction()
44 kernel_pack->GetJson()->contents + kernel_pack->GetJson()->len); in GetFunction()
74 …CUresult result = cuModuleLoadDataEx(&module, kernel_pack->GetKernel()->contents, 1, options, valu… in GetFunction()
88 GpuKernelMod::GpuKernelMod(const KernelPackPtr &kernel_pack) : kernel_pack_(kernel_pack) {} in GpuKernelMod() argument
|
| D | akg_gpu_kernel_build.cc | 39 void AkgGpuKernelBuilder::AkgSetKernelMod(const KernelPackPtr &kernel_pack, in AkgSetKernelMod() argument
41 auto kernel_mod_ptr = std::make_shared<GpuKernelMod>(kernel_pack); in AkgSetKernelMod()
42 auto kernel_json_info = kernel_pack->kernel_json_info(); in AkgSetKernelMod()
|
| D | akg_gpu_kernel_mod.h | 48 …CUresult GetFunction(const KernelPackPtr &kernel_pack, bool force_reload, std::vector<uint32_t> *t…
58 explicit GpuKernelMod(const KernelPackPtr &kernel_pack);
|
| D | akg_gpu_kernel_build.h | 33 …void AkgSetKernelMod(const KernelPackPtr &kernel_pack, const AkgKernelJsonGenerator &json_generato…
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/ |
| D | tbe_kernel_parallel_build.cc | 177 auto kernel_pack = TbeUtils::InsertCache(json_name, processor); in TaskFinishProcess() local
178 if (kernel_pack == nullptr) { in TaskFinishProcess()
189 …= GenKernelMod(task_iter->second.input_size_list, task_iter->second.output_size_list, kernel_pack); in TaskFinishProcess()
259 auto kernel_pack = TbeUtils::SearchCache(task_info.json_name); in GenSameFusionOpKernelMod() local
260 if (kernel_pack != nullptr) { in GenSameFusionOpKernelMod()
261 …uto kernel_mode = GenKernelMod(task_info.input_size_list, task_info.output_size_list, kernel_pack); in GenSameFusionOpKernelMod()
288 … const mindspore::kernel::KernelPackPtr &kernel_pack) const { in GenKernelMod()
289 MS_EXCEPTION_IF_NULL(kernel_pack); in GenKernelMod()
290 auto kernel_json_info = kernel_pack->kernel_json_info(); in GenKernelMod()
291 auto kernel_mod_ptr = std::make_shared<TbeKernelMod>(kernel_pack); in GenKernelMod()
|
| D | tbe_utils.cc | 277 uintptr_t KernelManager::GenFuncStub(const mindspore::kernel::KernelPack &kernel_pack, bool force_r… in GenFuncStub() argument
281 auto kernel = kernel_pack.GetKernel(); in GenFuncStub()
289 auto kernel_json_info = kernel_pack.kernel_json_info(); in GenFuncStub()
307 if (BinaryRegister((*kernel_pack.GetKernel()), &module, magic, dynamic_flag) != 0) { in GenFuncStub()
331 std::string KernelManager::GetStubFuncName(const KernelPackPtr &kernel_pack) { in GetStubFuncName() argument
332 MS_EXCEPTION_IF_NULL(kernel_pack); in GetStubFuncName()
333 auto kernel_json_info = kernel_pack->kernel_json_info(); in GetStubFuncName()
|
| D | tbe_kernel_mod.h | 31 explicit TbeKernelMod(KernelPackPtr kernel_pack) : kernel_pack_(std::move(kernel_pack)) {} in TbeKernelMod() argument
|
| D | tbe_utils.h | 79 …static uintptr_t GenFuncStub(const KernelPack &kernel_pack, bool force_reload, uint32_t *block_dim,
82 static std::string GetStubFuncName(const KernelPackPtr &kernel_pack);
|
| D | tbe_kernel_parallel_build.h | 63 const KernelPackPtr &kernel_pack) const;
|
| D | ascend_kernel_compile.cc | 679 auto kernel_pack = TbeUtils::SearchCache(json_name); in AscendFusionOpCompile() local
680 if (kernel_pack != nullptr) { in AscendFusionOpCompile()
681 … auto kernel_mod = build_manager_->GenKernelMod(input_size_list, output_size_list, kernel_pack); in AscendFusionOpCompile()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/akg/ascend/ |
| D | akg_ascend_kernel_build.cc | 37 void AkgAscendKernelBuilder::AkgSetKernelMod(const KernelPackPtr &kernel_pack, in AkgSetKernelMod() argument
39 auto kernel_mod_ptr = std::make_shared<AkgKernelMod>(kernel_pack); in AkgSetKernelMod()
40 auto kernel_json_info = kernel_pack->kernel_json_info(); in AkgSetKernelMod()
|
| D | akg_ascend_kernel_mod.cc | 38 AkgKernelMod::AkgKernelMod(const KernelPackPtr &kernel_pack) : kernel_pack_(kernel_pack) {} in AkgKernelMod() argument
|
| D | akg_ascend_kernel_build.h | 37 …void AkgSetKernelMod(const KernelPackPtr &kernel_pack, const AkgKernelJsonGenerator &json_generato…
|
| D | akg_ascend_kernel_mod.h | 29 explicit AkgKernelMod(const KernelPackPtr &kernel_pack);
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/ |
| D | kernel_fusion.cc | 95 auto kernel_pack = TbeUtils::SearchCache(json_name); in KernelFusion() local
96 …if (kernel_pack != nullptr && ((!offline_tune.empty() && offline_tune != "true") || tune_mode == "… in KernelFusion()
97 auto kernel_mod = build_manger->GenKernelMod(input_size_list, output_size_list, kernel_pack); in KernelFusion()
|
| D | common_utils.cc | 221 KernelPackPtr kernel_pack = std::make_shared<KernelPack>(); in SearchCache() local
223 if (!kernel_pack->ReadFromJsonFile(kernel_json, processor)) { in SearchCache()
227 return kernel_pack; in SearchCache()
245 KernelPackPtr kernel_pack = std::make_shared<KernelPack>(); in InsertCache() local
246 if (!kernel_pack->ReadFromJsonFile(kernel_json, processor)) { in InsertCache()
258 return kernel_pack; in InsertCache()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/akg/ |
| D | akg_kernel_build.h | 44 …virtual void AkgSetKernelMod(const KernelPackPtr &kernel_pack, const AkgKernelJsonGenerator &json_…
|