| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/akg/ |
| D | akg_kernel_metadata.cc | 31 auto op_info_ptr = mindspore::kernel::OpLib::FindOp(op_name, OpImplyType::kAKG); in AkgMetadataInfo() local
32 if (op_info_ptr == nullptr) { in AkgMetadataInfo()
36 if (!ParseMetadata(kernel_node, op_info_ptr, Processor(i), kernel_info_list)) { in AkgMetadataInfo()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/tbe_json/ |
| D | single_tbe_json_creator.cc | 75 auto op_info_ptr = mindspore::kernel::tbe::TbeDynamicShapeUtil::FindOp(op_name, anf_node); in GenDataJson() local
76 auto inputs_ptr = op_info_ptr->inputs_ptr(); in GenDataJson()
104 auto op_info_ptr = mindspore::kernel::tbe::TbeDynamicShapeUtil::FindOp(op_name, anf_node); in GenInputsJson() local
105 MS_EXCEPTION_IF_NULL(op_info_ptr); in GenInputsJson()
106 std::vector<OpIOInfoPtr> inputs_ptr = op_info_ptr->inputs_ptr(); in GenInputsJson()
241 auto op_info_ptr = mindspore::kernel::tbe::TbeDynamicShapeUtil::FindOp(op_name, anf_node); in GenOutputsJson() local
242 MS_EXCEPTION_IF_NULL(op_info_ptr); in GenOutputsJson()
244 auto outputs_ptr = op_info_ptr->outputs_ptr(); in GenOutputsJson()
370 …ctTbeJsonCreator::AttrsJsonPostProcessing(const AnfNodePtr &anf_node, const OpInfoPtr &op_info_ptr, in AttrsJsonPostProcessing() argument
373 MS_EXCEPTION_IF_NULL(op_info_ptr); in AttrsJsonPostProcessing()
|
| D | tbe_json_creator.cc | 349 auto op_info_ptr = tbe::TbeDynamicShapeUtil::FindOp(op_name, cnode); in GenAttrsDescJson() local
351 GenAttrsJson(cnode, op_info_ptr, &attrs_json); in GenAttrsDescJson()
373 auto op_info_ptr = tbe::TbeDynamicShapeUtil::FindOp(op_name, cnode); in GenComputeCommonJson() local
374 auto func_name = op_info_ptr->kernel_name(); in GenComputeCommonJson()
376 auto python_module_path = op_info_ptr->impl_path(); in GenComputeCommonJson()
384 (*compute_json)[kJDynamicCompileStatic] = op_info_ptr->dynamic_compile_static(); in GenComputeCommonJson()
556 …l TbeJsonCreator::AttrsJsonPostProcessing(const AnfNodePtr &anf_node, const OpInfoPtr &op_info_ptr, in AttrsJsonPostProcessing() argument
|
| D | tbe_json_creator.h | 82 …void GenAttrsJson(const AnfNodePtr &anf_node, const OpInfoPtr &op_info_ptr, nlohmann::json *attrs_…
85 virtual bool AttrsJsonPostProcessing(const AnfNodePtr &anf_node, const OpInfoPtr &op_info_ptr,
|
| D | fusion_tbe_json_creator.h | 39 bool AttrsJsonPostProcessing(const AnfNodePtr &anf_node, const OpInfoPtr &op_info_ptr,
|
| D | single_tbe_json_creator.h | 61 bool AttrsJsonPostProcessing(const AnfNodePtr &anf_node, const OpInfoPtr &op_info_ptr,
|
| D | fusion_tbe_json_creator.cc | 303 …ldTbeJsonCreator::AttrsJsonPostProcessing(const AnfNodePtr &anf_node, const OpInfoPtr &op_info_ptr, in AttrsJsonPostProcessing() argument
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/aicpu/ |
| D | aicpu_kernel_metadata.cc | 35 auto op_info_ptr = mindspore::kernel::OpLib::FindOp(op_name, OpImplyType::kAICPU); in AicpuMetadataInfo() local
36 if (op_info_ptr == nullptr) { in AicpuMetadataInfo()
45 if (!ParseMetadata(kernel_node, op_info_ptr, AICPU, kernel_info_list)) { in AicpuMetadataInfo()
|
| D | aicpu_kernel_build.cc | 155 auto op_info_ptr = mindspore::kernel::OpLib::FindOp(op_name, OpImplyType::kAICPU); in SetNodeAttr() local
156 MS_EXCEPTION_IF_NULL(op_info_ptr); in SetNodeAttr()
157 auto attrs_ptr = op_info_ptr->attrs_ptr(); in SetNodeAttr()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/tbe_kernel_select/ |
| D | tbe_kernel_select.cc | 63 auto op_info_ptr = tbe::TbeDynamicShapeUtil::FindOp(node_name_, cnode_ptr_); in TbeMetadataInfoEx() local
64 if (!op_info_ptr) { in TbeMetadataInfoEx()
72 if (op_info_ptr->is_dynamic_format()) { in TbeMetadataInfoEx()
73 GetDynamicFormatPatternKernelInfo(*op_info_ptr); in TbeMetadataInfoEx()
75 OpPattern pattern = op_info_ptr->op_pattern(); in TbeMetadataInfoEx()
77 GetCommonPatternKernelInfo(*op_info_ptr); in TbeMetadataInfoEx()
79 GetAgnosticPatternKernelInfo(*op_info_ptr); in TbeMetadataInfoEx()
81 GetBroadcastPatternKernelInfo(*op_info_ptr); in TbeMetadataInfoEx()
83 GetReducePatternKernelInfo(*op_info_ptr); in TbeMetadataInfoEx()
89 FilterInVaildKernelInfo(*op_info_ptr); in TbeMetadataInfoEx()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/ |
| D | kernel_info_setter.cc | 81 auto op_info_ptr = mindspore::kernel::OpLib::FindOp(op_name, kernel::OpImplyType::kAKG); in SupportedTypeList() local
82 if (op_info_ptr == nullptr) { in SupportedTypeList()
85 (void)ParseMetadata(kernel_node, op_info_ptr, kernel::Processor::CUDA, &kernel_info_list); in SupportedTypeList()
121 auto op_info_ptr = mindspore::kernel::OpLib::FindOp(op_name, kernel::OpImplyType::kAKG); in SelectAkgKernel() local
122 if (op_info_ptr == nullptr) { in SelectAkgKernel()
126 if (!ParseMetadata(kernel_node, op_info_ptr, kernel::Processor::CUDA, &kernel_info_list)) { in SelectAkgKernel()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/ |
| D | cpu_kernel_factory.cc | 128 auto op_info_ptr = mindspore::kernel::OpLib::FindOp(kernel_name, kernel::OpImplyType::kCPU); in CPUKernelAttrCheck() local
129 if (op_info_ptr == nullptr) { in CPUKernelAttrCheck()
133 SetKernelAttrs(op_info_ptr, &kernel_attrs); in CPUKernelAttrCheck()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/ |
| D | common_utils.cc | 410 const std::shared_ptr<const OpInfo> &op_info_ptr) { in SetKernelBuildInfo() argument
412 MS_EXCEPTION_IF_NULL(op_info_ptr); in SetKernelBuildInfo()
414 auto imply_type = op_info_ptr->imply_type(); in SetKernelBuildInfo()
416 std::string fusion_name = op_info_ptr->fusion_type(); in SetKernelBuildInfo()
429 bool ParseMetadata(const CNodePtr &kernel_node, const std::shared_ptr<const OpInfo> &op_info_ptr, P… in ParseMetadata() argument
435 std::vector<std::shared_ptr<OpIOInfo>> inputs = op_info_ptr->inputs_ptr(); in ParseMetadata()
436 std::vector<std::shared_ptr<OpIOInfo>> outputs = op_info_ptr->outputs_ptr(); in ParseMetadata()
452 SetKernelBuildInfo(builder, processor, op_info_ptr); in ParseMetadata()
476 SetKernelBuildInfo(builder, processor, op_info_ptr); in ParseMetadata()
489 SetKernelBuildInfo(builder, processor, op_info_ptr); in ParseMetadata()
|
| D | common_utils.h | 83 bool ParseMetadata(const CNodePtr &kernel_node, const std::shared_ptr<const OpInfo> &op_info_ptr, P…
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/executor/tiling/ |
| D | op_tiling_adapter.cc | 234 auto op_info_ptr = mindspore::kernel::tbe::TbeDynamicShapeUtil::FindOp(op_name_, node); in InitOpIoName() local
235 MS_EXCEPTION_IF_NULL(op_info_ptr); in InitOpIoName()
238 auto inputs_ptr = op_info_ptr->inputs_ptr(); in InitOpIoName()
262 auto outputs_ptr = op_info_ptr->outputs_ptr(); in InitOpIoName()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/cpu/ |
| D | kernel_select_cpu.cc | 321 auto op_info_ptr = mindspore::kernel::OpLib::FindOp(op_name, kernel::OpImplyType::kCPU); in SetKernelInfo() local
322 if (op_info_ptr == nullptr) { in SetKernelInfo()
326 kernel::CPUKernelFactory::GetInstance().SetKernelAttrs(op_info_ptr, &kernel_attrs); in SetKernelInfo()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/ |
| D | tbe_kernel_build.cc | 193 auto op_info_ptr = mindspore::kernel::tbe::TbeDynamicShapeUtil::FindOp(op_name, anf_node); in GenTbeSingleKernelJson() local
194 MS_EXCEPTION_IF_NULL(op_info_ptr); in GenTbeSingleKernelJson()
196 (*kernel_json)[kImplPath] = op_info_ptr->impl_path(); in GenTbeSingleKernelJson()
200 auto func_name = op_info_ptr->kernel_name(); in GenTbeSingleKernelJson()
206 if (!GenTbeInputsJson(anf_node, op_info_ptr, &inputs_json)) { in GenTbeSingleKernelJson()
213 if (!GenTbeOutputsJson(anf_node, op_info_ptr, &outputs_json)) { in GenTbeSingleKernelJson()
220 GenTbeAttrJson(anf_node, op_info_ptr, &attrs_json); in GenTbeSingleKernelJson()
1053 auto op_info_ptr = tbe::TbeDynamicShapeUtil::FindOp(origin_type, cnode); in GenFusionComputeCommonJson() local
1057 (*compute_op_str)[kJDynamicCompileStatic] = op_info_ptr->dynamic_compile_static(); in GenFusionComputeCommonJson()
1058 auto func_name = op_info_ptr->kernel_name(); in GenFusionComputeCommonJson()
[all …]
|