| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/ |
| D | kernel_build_info.cc | 22 std::string KernelBuildInfo::GetInputFormat(size_t input_index) const { in GetInputFormat()
30 std::string KernelBuildInfo::GetOutputFormat(size_t output_index) const { in GetOutputFormat()
38 TypeId KernelBuildInfo::GetInputDeviceType(size_t input_index) const { in GetInputDeviceType()
46 TypeId KernelBuildInfo::GetOutputDeviceType(size_t output_index) const { in GetOutputDeviceType()
54 const std::string &KernelBuildInfo::GetOriginDataFormat() const { return origin_data_format_; } in GetOriginDataFormat()
56 const std::vector<std::string> &KernelBuildInfo::GetAllInputFormats() const { return inputs_format_… in GetAllInputFormats()
58 const std::vector<std::string> &KernelBuildInfo::GetAllOutputFormats() const { return outputs_forma… in GetAllOutputFormats()
60 const std::vector<TypeId> &KernelBuildInfo::GetAllInputDeviceTypes() const { return inputs_device_t… in GetAllInputDeviceTypes()
62 const std::vector<TypeId> &KernelBuildInfo::GetAllOutputDeviceTypes() const { return outputs_device… in GetAllOutputDeviceTypes()
64 size_t KernelBuildInfo::GetInputNum() const { return inputs_format_.size(); } in GetInputNum()
[all …]
|
| D | kernel_build_info.h | 30 class KernelBuildInfo {
34 KernelBuildInfo() { in KernelBuildInfo() function
49 ~KernelBuildInfo() = default;
99 bool IsSimilarityKernelBuildInfo(const KernelBuildInfo &other) const;
101 bool operator==(const KernelBuildInfo &other) const;
103 bool operator!=(const KernelBuildInfo &other) const;
122 using KernelBuildInfoPtr = std::shared_ptr<KernelBuildInfo>;
124 class KernelBuildInfo::KernelBuildInfoBuilder {
126 KernelBuildInfoBuilder() { kernel_build_info_ = std::make_shared<KernelBuildInfo>(); } in KernelBuildInfoBuilder()
128 explicit KernelBuildInfoBuilder(std::shared_ptr<KernelBuildInfo> kernel_build_info) in KernelBuildInfoBuilder()
[all …]
|
| D | kernel_query.cc | 32 … std::vector<std::shared_ptr<kernel::KernelBuildInfo>> *kernel_info_list) { in FilterInvalidKernelInfo()
40 std::vector<std::shared_ptr<kernel::KernelBuildInfo>> filtered_list; in FilterInvalidKernelInfo()
43 …[output_tensor_num, input_tensor_num](const std::shared_ptr<kernel::KernelBuildInfo> &kernel_build… in FilterInvalidKernelInfo()
71 void CheckKernelInfoListEmpty(const std::vector<std::shared_ptr<kernel::KernelBuildInfo>> *kernel_i… in CheckKernelInfoListEmpty()
80 std::vector<std::shared_ptr<kernel::KernelBuildInfo>> *kernel_info_list) { in KernelQueryAll()
98 void KernelQuery(const CNodePtr &kernel_node, std::vector<std::shared_ptr<kernel::KernelBuildInfo>>… in KernelQuery()
128 void AICPUQuery(const CNodePtr &kernel_node, std::vector<std::shared_ptr<kernel::KernelBuildInfo>> … in AICPUQuery()
139 std::vector<std::shared_ptr<kernel::KernelBuildInfo>> kernel_info_list; in IsSupportedByAICPU()
153 std::vector<std::shared_ptr<kernel::KernelBuildInfo>> kernel_info_list; in IsSupportedByAICore()
|
| D | kernel_query.h | 28 void KernelQuery(const CNodePtr &kernel_node, std::vector<std::shared_ptr<kernel::KernelBuildInfo>>…
30 void AICPUQuery(const CNodePtr &kernel_node, std::vector<std::shared_ptr<kernel::KernelBuildInfo>> …
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/ |
| D | kernel_select_ascend.cc | 58 bool MatchInferOutputDataType(const CNodePtr &cnode, const kernel::KernelBuildInfo &kernel_build_in… in MatchInferOutputDataType()
136 void UpdateCurMatchCounts(const kernel::KernelBuildInfo &kernel_build_info, const std::shared_ptr<C… in UpdateCurMatchCounts()
187 const CNodePtr &cnode, const std::shared_ptr<kernel::KernelBuildInfo> &selected_kernel_build_info, in PrintRaiseOrReducePrecisionSelectedInfo()
203 std::shared_ptr<kernel::KernelBuildInfo> ChooseMatchedKernelInfo( in ChooseMatchedKernelInfo()
204 …const CNodePtr &kernel_node, const std::vector<std::shared_ptr<kernel::KernelBuildInfo>> &kernel_i… in ChooseMatchedKernelInfo()
223 std::vector<std::shared_ptr<kernel::KernelBuildInfo>> FilteredKernelInfoByDtype( in FilteredKernelInfoByDtype()
224 …const CNodePtr &cnode, const std::vector<std::shared_ptr<kernel::KernelBuildInfo>> &kernel_info_li… in FilteredKernelInfoByDtype()
225 std::vector<std::shared_ptr<kernel::KernelBuildInfo>> result; in FilteredKernelInfoByDtype()
254 bool TagRaiseReduce(const std::shared_ptr<kernel::KernelBuildInfo> &kernel_build_info, const CNodeP… in TagRaiseReduce()
289 std::vector<std::shared_ptr<kernel::KernelBuildInfo>> FilterRaisedOrReducePrecisionMatchedKernelInf… in FilterRaisedOrReducePrecisionMatchedKernelInfo()
[all …]
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/tbe_kernel_select/ |
| D | tbe_kernel_select.h | 29 void TbeMetadataInfo(const CNodePtr &kernel_node, std::vector<std::shared_ptr<KernelBuildInfo>> *ke…
33 using KernelBuildInfoIter = std::vector<std::shared_ptr<KernelBuildInfo>>::iterator;
36 …TbeKernelSelect(CNodePtr kernel_node, std::vector<std::shared_ptr<KernelBuildInfo>> *kernel_info_l…
50 …static void SetTbeBuildCommonInfo(const OpInfo &op_info, KernelBuildInfo::KernelBuildInfoBuilder *…
72 std::vector<std::shared_ptr<KernelBuildInfo>> *kernel_info_list_;
|
| /third_party/mindspore/tests/ut/cpp/pre_activate/ascend/format_type/ |
| D | check_consistency_test.cc | 66 kernel::KernelBuildInfo::KernelBuildInfoBuilder builder1; in TEST_F()
79 kernel::KernelBuildInfo::KernelBuildInfoBuilder builder2; in TEST_F()
92 kernel::KernelBuildInfo::KernelBuildInfoBuilder builder3; in TEST_F()
127 kernel::KernelBuildInfo::KernelBuildInfoBuilder builder1; in TEST_F()
140 kernel::KernelBuildInfo::KernelBuildInfoBuilder builder2; in TEST_F()
153 kernel::KernelBuildInfo::KernelBuildInfoBuilder builder3; in TEST_F()
|
| D | merge_cast_to_op_test.cc | 47 std::vector<std::shared_ptr<kernel::KernelBuildInfo>> *kernel_info_list) override { in Query()
50 kernel::KernelBuildInfo::KernelBuildInfoBuilder builder; in Query()
57 kernel::KernelBuildInfo::KernelBuildInfoBuilder builder; in Query()
96 kernel::KernelBuildInfo::KernelBuildInfoBuilder builder1; in TEST_F()
109 kernel::KernelBuildInfo::KernelBuildInfoBuilder builder2; in TEST_F()
159 kernel::KernelBuildInfo::KernelBuildInfoBuilder builder1; in TEST_F()
172 kernel::KernelBuildInfo::KernelBuildInfoBuilder builder2; in TEST_F()
|
| D | insert_cast_test.cc | 56 kernel::KernelBuildInfo::KernelBuildInfoBuilder builder; in TEST_F()
64 kernel::KernelBuildInfo::KernelBuildInfoBuilder builder1; in TEST_F()
113 kernel::KernelBuildInfo::KernelBuildInfoBuilder builder; in TEST_F()
118 kernel::KernelBuildInfo::KernelBuildInfoBuilder builder1; in TEST_F()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/ |
| D | kernel_info_setter.cc | 37 using mindspore::kernel::KernelBuildInfo;
39 bool CheckKernelInfo(const std::shared_ptr<KernelBuildInfo> &alternative_kernel_info, in CheckKernelInfo()
40 const std::shared_ptr<KernelBuildInfo> &selected_kernel_info) { in CheckKernelInfo()
79 std::vector<std::shared_ptr<KernelBuildInfo>> kernel_info_list; in SupportedTypeList()
103 bool SelectAkgKernel(const CNodePtr &kernel_node, const std::shared_ptr<KernelBuildInfo> &selected_… in SelectAkgKernel()
106 std::vector<std::shared_ptr<KernelBuildInfo>> kernel_info_list; in SelectAkgKernel()
134 [&](const std::shared_ptr<KernelBuildInfo> &alternative_kernel_info) { in SelectAkgKernel()
144 void SetTensorDeviceInfo(const kernel::KernelBuildInfo &selected_kernel_info, const CNodePtr &kerne… in SetTensorDeviceInfo()
156 std::shared_ptr<kernel::KernelBuildInfo::KernelBuildInfoBuilder> builder = in SetTensorDeviceInfo()
157 std::make_shared<kernel::KernelBuildInfo::KernelBuildInfoBuilder>(); in SetTensorDeviceInfo()
[all …]
|
| /third_party/mindspore/tests/ut/cpp/pre_activate/ascend/ir_fusion/ |
| D | layer_norm_beta_gamma_backprop_fusion_test.cc | 44 …void Query(const CNodePtr &cnode, std::vector<std::shared_ptr<kernel::KernelBuildInfo>> *kernel_in… in Query()
46 kernel::KernelBuildInfo::KernelBuildInfoBuilder builder; in Query()
76 kernel::KernelBuildInfo::KernelBuildInfoBuilder builder1; in TEST_F()
95 kernel::KernelBuildInfo::KernelBuildInfoBuilder builder2; in TEST_F()
137 kernel::KernelBuildInfo::KernelBuildInfoBuilder builder1; in TEST_F()
154 kernel::KernelBuildInfo::KernelBuildInfoBuilder builder2; in TEST_F()
196 kernel::KernelBuildInfo::KernelBuildInfoBuilder builder1; in TEST_F()
213 kernel::KernelBuildInfo::KernelBuildInfoBuilder builder2; in TEST_F()
250 kernel::KernelBuildInfo::KernelBuildInfoBuilder builder1; in TEST_F()
262 kernel::KernelBuildInfo::KernelBuildInfoBuilder builder2; in TEST_F()
[all …]
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/format_type/ |
| D | merge_cast_to_op.cc | 32 … const std::shared_ptr<kernel::KernelBuildInfo> &candidate_kernel_info) { in AlternativeKernelInfoForInput()
73 bool CheckInputs(const CNodePtr &node, const std::shared_ptr<kernel::KernelBuildInfo> &kernel_info)… in CheckInputs()
88 bool CheckOtherOutputs(const CNodePtr &node, const std::shared_ptr<kernel::KernelBuildInfo> &kernel… in CheckOtherOutputs()
106 bool CheckIndexOutput(const CNodePtr &node, const std::shared_ptr<kernel::KernelBuildInfo> &kernel_… in CheckIndexOutput()
164 std::vector<std::shared_ptr<kernel::KernelBuildInfo>> kernel_info_list; in MergeCastToNextOp()
170 …[&next_cnode, &dst_type_id, &cast_index](const std::shared_ptr<kernel::KernelBuildInfo> &candidate… in MergeCastToNextOp()
234 std::vector<std::shared_ptr<kernel::KernelBuildInfo>> kernel_info_list; in MergeCastToPriorOp()
242 …[&prior_op, &cur_node, &output_idx](const std::shared_ptr<kernel::KernelBuildInfo> &item_kernel_in… in MergeCastToPriorOp()
|
| D | dynamic_rnn_grad_reformat.cc | 50 auto matmul_new_builder = std::make_shared<kernel::KernelBuildInfo::KernelBuildInfoBuilder>(); in Process()
64 auto split_new_builder = std::make_shared<kernel::KernelBuildInfo::KernelBuildInfoBuilder>(); in Process()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/rts/ |
| D | rt_kernel_info.h | 38 virtual std::vector<std::shared_ptr<kernel::KernelBuildInfo>> GetKernelInfo() { in GetKernelInfo()
39 return std::vector<std::shared_ptr<kernel::KernelBuildInfo>>{}; in GetKernelInfo()
71 void GetRtKelInfo(const CNodePtr &kernel_node, std::vector<std::shared_ptr<kernel::KernelBuildInfo>…
|
| D | label_switch.cc | 75 std::vector<std::shared_ptr<kernel::KernelBuildInfo>> LabelSwitchDesc::GetKernelInfo() { in GetKernelInfo()
76 std::vector<std::shared_ptr<kernel::KernelBuildInfo>> label_switch_build_info{}; in GetKernelInfo()
84 auto builder = KernelBuildInfo::KernelBuildInfoBuilder(); in GetKernelInfo()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/ |
| D | gpu_kernel_factory.h | 48 … std::shared_ptr<mindspore::kernel::KernelBuildInfo::KernelBuildInfoBuilder> builder);
59 …std::pair<bool, size_t> GpuKernelAttrCheck(const std::string &kernel_name, const KernelBuildInfo *…
60 void CheckSM(const KernelBuildInfo *kernel_info, const size_t &input_index);
61 bool CheckIOParam(const std::string &kernel_name, const KernelBuildInfo *kernel_info,
|
| D | gpu_kernel_factory.cc | 39 bool GpuKernelFactory::CheckIOParam(const std::string &kernel_name, const KernelBuildInfo *kernel_i… in CheckIOParam()
79 …const std::string &kernel_name, std::shared_ptr<mindspore::kernel::KernelBuildInfo::KernelBuildInf… in ReducePrecision()
112 void GpuKernelFactory::CheckSM(const KernelBuildInfo *kernel_info, const size_t &input_index) { in CheckSM()
127 const KernelBuildInfo *kernel_info) { in GpuKernelAttrCheck()
183 const KernelBuildInfo *kernel_build_Info = kernel_info->select_kernel_build_info(); in Create()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/aicpu/ |
| D | aicpu_kernel_metadata.cc | 27 void AicpuMetadataInfo(const CNodePtr &kernel_node, std::vector<std::shared_ptr<KernelBuildInfo>> *… in AicpuMetadataInfo()
52 … std::vector<std::shared_ptr<KernelBuildInfo>> *kernel_info_list) { in AicpuMetadataInfoForSpecialNodes()
71 auto builder = KernelBuildInfo::KernelBuildInfoBuilder(); in AicpuMetadataInfoForSpecialNodes()
|
| D | aicpu_kernel_metadata.h | 27 void AicpuMetadataInfo(const CNodePtr &kernel_node, std::vector<std::shared_ptr<KernelBuildInfo>> *…
29 … std::vector<std::shared_ptr<KernelBuildInfo>> *kernel_info_list);
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fusion/ |
| D | transposed_update_fusion.cc | 58 kernel::KernelBuildInfo::KernelBuildInfoBuilder op_builder; in CreatePermValueNode()
86 std::vector<std::shared_ptr<kernel::KernelBuildInfo>> kernel_info_list; in Process()
95 auto builder = std::make_shared<kernel::KernelBuildInfo::KernelBuildInfoBuilder>(ori_build_info); in Process()
|
| D | refresh_parameter_format.cc | 32 std::shared_ptr<kernel::KernelBuildInfo::KernelBuildInfoBuilder> builder = in DoRefresh()
33 std::make_shared<kernel::KernelBuildInfo::KernelBuildInfoBuilder>(); in DoRefresh()
|
| /third_party/mindspore/tests/ut/cpp/pre_activate/ascend/ir_fission/ |
| D | bn_grad_split_test.cc | 75 kernel::KernelBuildInfo::KernelBuildInfoBuilder builder2; in TEST_F()
81 kernel::KernelBuildInfo::KernelBuildInfoBuilder builder1; in TEST_F()
141 kernel::KernelBuildInfo::KernelBuildInfoBuilder builder2; in TEST_F()
147 kernel::KernelBuildInfo::KernelBuildInfoBuilder builder1; in TEST_F()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/host/ |
| D | host_kernel_metadata.cc | 30 void HostMetadataInfo(const CNodePtr &kernel_node, std::vector<std::shared_ptr<KernelBuildInfo>> *k… in HostMetadataInfo()
55 auto builder = KernelBuildInfo::KernelBuildInfoBuilder(); in HostMetadataInfo()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/ |
| D | decrease_transfer_precision.cc | 166 kernel::KernelBuildInfo::KernelBuildInfoBuilder graph_info_builder; in Process_Father()
195 …std::make_shared<kernel::KernelBuildInfo::KernelBuildInfoBuilder>(AnfAlgo::GetSelectKernelBuildInf… in Process_Father()
222 …auto gk_builder_info = std::make_shared<kernel::KernelBuildInfo::KernelBuildInfoBuilder>(old_build… in Process_Father()
247 …std::make_shared<kernel::KernelBuildInfo::KernelBuildInfoBuilder>(AnfAlgo::GetSelectKernelBuildInf… in Process_Son()
279 kernel::KernelBuildInfo::KernelBuildInfoBuilder node_info_builder; in Process_Son()
|
| /third_party/mindspore/tests/ut/cpp/pre_activate/pass/ |
| D | allreduce_fusion_test.cc | 52 kernel::KernelBuildInfo::KernelBuildInfoBuilder builder; in TEST_F()
93 kernel::KernelBuildInfo::KernelBuildInfoBuilder builder; in TEST_F()
134 kernel::KernelBuildInfo::KernelBuildInfoBuilder builder; in TEST_F()
191 kernel::KernelBuildInfo::KernelBuildInfoBuilder builder; in TEST_F()
|