| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/ |
| D | kernel_build_info.h | 128 explicit KernelBuildInfoBuilder(std::shared_ptr<KernelBuildInfo> kernel_build_info) in KernelBuildInfoBuilder() argument
130 SetKernelType(kernel_build_info->kernel_type()); in KernelBuildInfoBuilder()
131 SetFusionType(kernel_build_info->fusion_type()); in KernelBuildInfoBuilder()
132 SetProcessor(kernel_build_info->processor()); in KernelBuildInfoBuilder()
133 SetOpPattern(kernel_build_info->op_pattern()); in KernelBuildInfoBuilder()
134 for (size_t index = 0; index < kernel_build_info->GetInputNum(); ++index) { in KernelBuildInfoBuilder()
135 …kernel_build_info_->inputs_device_type_.emplace_back(kernel_build_info->GetInputDeviceType(index)); in KernelBuildInfoBuilder()
136 kernel_build_info_->inputs_format_.emplace_back(kernel_build_info->GetInputFormat(index)); in KernelBuildInfoBuilder()
137 …kernel_build_info_->input_reshape_type_.emplace_back(kernel_build_info->GetInputReshapeType(index)… in KernelBuildInfoBuilder()
138 …kernel_build_info_->input_value_depend_.emplace_back(kernel_build_info->GetInputValueDepend(index)… in KernelBuildInfoBuilder()
[all …]
|
| D | kernel_query.cc | 43 …_tensor_num, input_tensor_num](const std::shared_ptr<kernel::KernelBuildInfo> &kernel_build_info) { in FilterInvalidKernelInfo() argument
44 return kernel_build_info->GetOutputNum() == output_tensor_num && in FilterInvalidKernelInfo()
45 kernel_build_info->GetInputNum() == input_tensor_num; in FilterInvalidKernelInfo()
|
| D | CMakeLists.txt | 2 "kernel_build_info.cc"
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/ |
| D | kernel_select_ascend.cc | 58 …MatchInferOutputDataType(const CNodePtr &cnode, const kernel::KernelBuildInfo &kernel_build_info) { in MatchInferOutputDataType() argument
61 for (size_t input_index = 0; input_index < kernel_build_info.GetInputNum(); ++input_index) { in MatchInferOutputDataType()
63 if (kernel_build_info.GetInputDeviceType(input_index) != input_origin_type) { in MatchInferOutputDataType()
68 for (size_t output_index = 0; output_index < kernel_build_info.GetOutputNum(); ++output_index) { in MatchInferOutputDataType()
69 …if (kernel_build_info.GetOutputDeviceType(output_index) != AnfAlgo::GetOutputInferDataType(cnode, … in MatchInferOutputDataType()
136 void UpdateCurMatchCounts(const kernel::KernelBuildInfo &kernel_build_info, const std::shared_ptr<C… in UpdateCurMatchCounts() argument
153 …if (kernel_build_info.GetInputFormat(input_index) == AnfAlgo::GetPrevNodeOutputFormat(kernel_node,… in UpdateCurMatchCounts()
161 if (kernel_build_info.GetInputDeviceType(input_index) == prev_device_type) { in UpdateCurMatchCounts()
164 if (kernel_build_info.GetInputFormat(input_index) == pri_match_format) { in UpdateCurMatchCounts()
167 if (kernel_build_info.GetInputFormat(input_index) == kOpFormat_DEFAULT || in UpdateCurMatchCounts()
[all …]
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/gpu/ |
| D | combine_momentum_fusion.cc | 94 auto kernel_build_info = GenerateKernelBuildInfo(momentums); in Run() local
95 AnfAlgo::SetSelectKernelBuildInfo(kernel_build_info, combine_mom.get()); in Run()
|
| D | combine_cast_fusion.cc | 111 auto kernel_build_info = GenerateKernelBuildInfo(cast_list); in Run() local
112 AnfAlgo::SetSelectKernelBuildInfo(kernel_build_info, cast_all.get()); in Run()
|
| D | concat_outputs_for_all_gather.cc | 111 auto kernel_build_info = GenerateKernelBuildInfo(concat, output_info, inputs_size, i); in InsertConcatForOutput() local
112 AnfAlgo::SetSelectKernelBuildInfo(kernel_build_info, concat.get()); in InsertConcatForOutput()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/pass/ |
| D | replace_node_by_proxy.cc | 84 auto kernel_build_info = GenerateKernelBuildInfo(cnode); in Run() local
85 AnfAlgo::SetSelectKernelBuildInfo(kernel_build_info, proxy_node.get()); in Run()
|
| D | communication_op_fusion.cc | 369 auto kernel_build_info = GenerateKernelBuildInfo(communication_op_info, start_index, end_index); in CreateFusedCommunicationOp() local
370 AnfAlgo::SetSelectKernelBuildInfo(kernel_build_info, fused_node.get()); in CreateFusedCommunicationOp()
|
| /third_party/mindspore/mindspore/ccsrc/backend/session/ |
| D | ascend_inference_session.cc | 161 auto kernel_build_info = AnfAlgo::GetSelectKernelBuildInfo(parameter); in CompareInput() local
162 if (input->data_type() != kernel_build_info->GetOutputDeviceType(0)) { in CompareInput()
164 … << ", but the parameter data type is " << kernel_build_info->GetOutputDeviceType(0) in CompareInput()
|
| D | gpu_inference_session.cc | 161 auto kernel_build_info = AnfAlgo::GetSelectKernelBuildInfo(parameter); in CompareInput() local
162 if (input->data_type() != kernel_build_info->GetOutputDeviceType(0)) { in CompareInput()
164 … << ", but the parameter data type is " << kernel_build_info->GetOutputDeviceType(0) in CompareInput()
|
| D | session_basic.cc | 1775 auto kernel_build_info = AnfAlgo::GetSelectKernelBuildInfo(parameter); in GetModelInputsInfo() local
1776 auto data_type = kernel_build_info->GetOutputDeviceType(0); in GetModelInputsInfo()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/ |
| D | gpu_kernel_factory.cc | 192 …ry::SearchRegistered(const std::string &kernel_name, const KernelBuildInfoPtr &kernel_build_info) { in SearchRegistered() argument
193 std::pair<bool, size_t> ret_pair = GpuKernelAttrCheck(kernel_name, kernel_build_info.get()); in SearchRegistered()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/buffer_fusion/ |
| D | fusion_base_pass.h | 55 kernel::KernelBuildInfoPtr kernel_build_info; member
|
| D | ub_pattern_fusion.cc | 440 buffer_fusion_info.second.kernel_build_info = in GetBufferFusionInfo()
509 AnfAlgo::SetSelectKernelBuildInfo(buffer_fusion_info.kernel_build_info, buffer_fusion.get()); in ReplaceFusionOp()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/enhancer/ |
| D | concat_outputs_for_all_gather.cc | 113 auto kernel_build_info = GenerateKernelBuildInfo(concat, output_info, inputs_size, i); in InsertConcatForOutput() local
114 AnfAlgo::SetSelectKernelBuildInfo(kernel_build_info, concat.get()); in InsertConcatForOutput()
|
| /third_party/mindspore/mindspore/ccsrc/ps/ |
| D | util.cc | 182 auto kernel_build_info = GenerateKernelBuildInfo(single_nodes); in DoFusion() local
183 AnfAlgo::SetSelectKernelBuildInfo(kernel_build_info, fused_cnode.get()); in DoFusion()
|