Home
last modified time | relevance | path

Searched refs:op_info (Results 1 – 25 of 95) sorted by relevance

1234

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/oplib/
Doplib.cc111 void OpLib::DecodeTBESpecificInfo(const nlohmann::json &obj, const std::shared_ptr<OpInfo> &op_info… in DecodeTBESpecificInfo() argument
114 MS_EXCEPTION_IF_NULL(op_info); in DecodeTBESpecificInfo()
115 op_info->set_async_flag(obj.at(kAsyncFlag)); in DecodeTBESpecificInfo()
116 op_info->set_binfile_name(obj.at(kBinfileName)); in DecodeTBESpecificInfo()
117 op_info->set_compute_cost(obj.at(kComputeCost)); in DecodeTBESpecificInfo()
118 op_info->set_kernel_name(obj.at(kKernelName)); in DecodeTBESpecificInfo()
119 op_info->set_partial_flag(obj.at(kPartialFlag)); in DecodeTBESpecificInfo()
120 op_info->set_need_check_supported(obj.at(kNeedCheckSupported)); in DecodeTBESpecificInfo()
123 op_info->set_dynamic_shape(obj.at(kDynamicShape)); in DecodeTBESpecificInfo()
127 op_info->set_dynamic_compile_static_(obj.at(kDynamicCompileStatic)); in DecodeTBESpecificInfo()
[all …]
Doplib.h43 …odeAttr(const nlohmann::json &obj, OpImplyType imply_type, const std::shared_ptr<OpInfo> &op_info);
46 …atic void DecodeTBESpecificInfo(const nlohmann::json &obj, const std::shared_ptr<OpInfo> &op_info);
47 …atic void DecodeAKGSpecificInfo(const nlohmann::json &obj, const std::shared_ptr<OpInfo> &op_info);
49 … const std::shared_ptr<OpInfo> &op_info, const nlohmann::json &dtype_format);
50 static bool GetRefInfo(const std::shared_ptr<OpInfo> &op_info);
51 static bool CheckRepetition(const std::shared_ptr<OpInfo> &op_info);
/third_party/mindspore/tests/ut/python/dataset/
Dtest_profiling.py68 op_info = data["op_info"]
70 assert len(op_info) == num_pipeline_ops + 1
79 op_info = data["op_info"]
81 assert len(op_info) == num_ops
83 assert op_info[i]["op_type"] in op_list
147 op_info = data["op_info"]
148 assert len(op_info) == 5
150 if op_info[i]["op_type"] != "ZipOp":
151 assert "size" in op_info[i]["metrics"]["output_queue"]
152 assert "length" in op_info[i]["metrics"]["output_queue"]
[all …]
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/tbe_kernel_select/
Dtbe_kernel_select.cc92 void TbeKernelSelect::GetCommonPatternKernelInfo(const OpInfo &op_info) { in GetCommonPatternKernelInfo() argument
96 const auto inputs_info = op_info.inputs_ptr(); in GetCommonPatternKernelInfo()
98 const auto outputs_info = op_info.outputs_ptr(); in GetCommonPatternKernelInfo()
107 SetTbeBuildCommonInfo(op_info, &builder); in GetCommonPatternKernelInfo()
137 void TbeKernelSelect::GetDynamicFormatPatternKernelInfo(const OpInfo &op_info) { in GetDynamicFormatPatternKernelInfo() argument
139 CreateNewOpInfo(op_info, &op_info_new); in GetDynamicFormatPatternKernelInfo()
143 void TbeKernelSelect::GetAgnosticPatternKernelInfo(const OpInfo &op_info) { in GetAgnosticPatternKernelInfo() argument
144 if (op_info.inputs_ptr().size() != 1) { in GetAgnosticPatternKernelInfo()
155 input_item.assign(op_info.inputs_ptr().size(), format); in GetAgnosticPatternKernelInfo()
156 output_item.assign(op_info.outputs_ptr().size(), format); in GetAgnosticPatternKernelInfo()
[all …]
Dtbe_kernel_select.h41 void GetCommonPatternKernelInfo(const OpInfo &op_info);
42 void GetDynamicFormatPatternKernelInfo(const OpInfo &op_info);
43 void GetAgnosticPatternKernelInfo(const OpInfo &op_info);
44 void GetBroadcastPatternKernelInfo(const OpInfo &op_info);
45 void GetReducePatternKernelInfo(const OpInfo &op_info);
46 void FilterInVaildKernelInfo(const OpInfo &op_info);
50 …static void SetTbeBuildCommonInfo(const OpInfo &op_info, KernelBuildInfo::KernelBuildInfoBuilder *…
57 …static void CreateNewOpInfo(const OpInfo &op_info, const SupportFormat &support_format, OpInfo *op…
62 …void CreateNewOpInfo(const mindspore::kernel::OpInfo &op_info, mindspore::kernel::OpInfo *op_info_…
/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/executor/
Dhccl_dynamic_kernel.cc94 ::HcomOperation op_info; in Execute()
95 op_info.hcclType = hccl_type_; in Execute()
96 op_info.inputPtr = input_ptr_; in Execute()
97 op_info.outputPtr = output_ptr_; in Execute()
98 op_info.dataType = static_cast<HcclDataType>(data_type_); in Execute()
99 op_info.opType = static_cast<HcclReduceOp>(op_type_); in Execute()
100 op_info.root = IntToUint(root_); in Execute()
101 op_info.count = count_; in Execute()
112 auto hccl_ret = hccl::HcclAdapter::GetInstance().HcclExecEnqueueOp(op_info, callback); in Execute()
/third_party/mindspore/scripts/map_dump_file_to_code/
Dmap_file_to_code.py125 for _, op_info in self.ir_info_dict.items():
126 if op_info["op_num"] == dump_op_num and op_info["in_file"] is not None:
127 if dump_op_name in (dump_op_num, op_info["op_name"]):
128 if not op_info["in_file"]:
129 … print("[WARNING] Cannot find {}'s source code in ir file.".format(op_info["origin_op_name"]))
131 print("[INFO] Find operation '{}'.".format(op_info["origin_op_name"]))
132 for line in op_info["in_file"]:
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/format_type/
Ddeal_ref_and_split_unsupported_transdata.cc38 auto op_info = mindspore::kernel::tbe::TbeDynamicShapeUtil::FindOp(op_name, cnode); in FindRefOriginNode() local
40 if (op_info != nullptr && op_info->is_ref()) { in FindRefOriginNode()
41 auto ref_infos = op_info->ref_infos(); in FindRefOriginNode()
166 …tr &func_graph, const CNodePtr &orig_cnode, const std::shared_ptr<kernel::OpInfo> &op_info) const { in DealRefForMultipleOutput()
182 MS_EXCEPTION_IF_NULL(op_info); in DealRefForMultipleOutput()
183 auto ref_infos = op_info->ref_infos(); in DealRefForMultipleOutput()
207 …raphPtr &func_graph, const CNodePtr &cnode, const std::shared_ptr<kernel::OpInfo> &op_info) const { in DealRefSingleOutput()
209 MS_EXCEPTION_IF_NULL(op_info); in DealRefSingleOutput()
210 auto ref_infos = op_info->ref_infos(); in DealRefSingleOutput()
257 auto op_info = mindspore::kernel::tbe::TbeDynamicShapeUtil::FindOp(op_name, cnode); in Process() local
[all …]
/third_party/mindspore/mindspore/ccsrc/ps/ps_cache/ascend/
Dascend_ps_cache.cc91 bool SetNodedefProto(const std::shared_ptr<KernelNodeInfo> &op_info, in SetNodedefProto() argument
93 MS_ERROR_IF_NULL(op_info); in SetNodedefProto()
96 proto.set_op(op_info->op_name_); in SetNodedefProto()
97 RETURN_IF_FALSE(SetProtoInputs(op_info->input_data_shape_, op_info->input_data_type_, &proto)); in SetNodedefProto()
98 RETURN_IF_FALSE(SetProtoOutputs(op_info->output_data_shape_, op_info->output_data_type_, &proto)); in SetNodedefProto()
104 MS_LOG(DEBUG) << "Set node def proto, node name:" << op_info->op_name_; in SetNodedefProto()
231 auto op_info = in HashSwapOut() local
233 MS_ERROR_IF_NULL_W_RET_VAL(op_info, false); in HashSwapOut()
234 RETURN_IF_FALSE(SetNodedefProto(op_info, hash_swap_out_mod)); in HashSwapOut()
272 auto op_info = in HashSwapIn() local
[all …]
/third_party/mindspore/mindspore/ccsrc/profiler/device/
Ddata_saver.cc25 OpDetailInfo::OpDetailInfo(const std::shared_ptr<OpInfo> op_info, float proportion) in OpDetailInfo() argument
26 : op_info_(op_info), proportion_(proportion) { in OpDetailInfo()
28 op_full_name_ = op_info->op_name; in OpDetailInfo()
33 if (op_info->op_count == 0) { in OpDetailInfo()
37 op_avg_time_ = op_info->op_host_cost_time / op_info->op_count; in OpDetailInfo()
50 auto op_info = std::make_shared<OpInfo>(item.second); in ParseOpInfo() local
51 if (op_info == nullptr) { in ParseOpInfo()
55 OpDetailInfo op_detail_info = OpDetailInfo(op_info, proportion); in ParseOpInfo()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/akg/
Dakg_kernel_json_generator.cc51 auto op_info = std::make_shared<OpInfo>(); in Run() local
52 op_info->set_op_name(AnfAlgo::GetCNodeName(cnode_)); in Run()
53 op_info->set_imply_type(OpImplyType::kAKG); in Run()
54 ExtractInputs(op_info); in Run()
55 ExtractOutputs(op_info); in Run()
56 ExtractAttrs(op_info); in Run()
57 return op_info; in Run()
61 void ExtractInputs(const OpInfoPtr &op_info) const { in ExtractInputs()
67 op_info->add_inputs_ptr(io_info); in ExtractInputs()
74 op_info->add_inputs_ptr(io_info); in ExtractInputs()
[all …]
/third_party/mindspore/mindspore/ccsrc/profiler/device/cpu/
Dcpu_profiling.cc52 OpInfo op_info; in SetRunTimeData() local
53 op_info.op_name = op_name; in SetRunTimeData()
54 op_info.pid = pid; in SetRunTimeData()
55 op_info.op_count = 1; in SetRunTimeData()
56 op_info_map_[op_name] = op_info; in SetRunTimeData()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/
Dascend_helper.h68 …auto op_info = mindspore::kernel::tbe::TbeDynamicShapeUtil::FindOp(AnfAlgo::GetCNodeName(node), no… in IsTbeRef() local
69 if (op_info != nullptr) { in IsTbeRef()
70 return op_info->is_ref(); in IsTbeRef()
93 auto op_info = kernel::tbe::TbeDynamicShapeUtil::FindOp(op_name, cnode); in GetOpRegisteredOutputNum() local
94 if (op_info == nullptr) { in GetOpRegisteredOutputNum()
97 return op_info->outputs_ptr().size(); in GetOpRegisteredOutputNum()
/third_party/mindspore/mindspore/_extends/parallel_compile/tbe_compiler/
Dtbe_common.py124 def get_args(op_info, arg_type): argument
135 if arg_type not in op_info:
138 if not op_info[arg_type]:
141 arg_info = op_info[arg_type]
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/
Dcpu_kernel_factory.cc58 void CPUKernelFactory::SetKernelAttrs(const std::shared_ptr<kernel::OpInfo> op_info, in SetKernelAttrs() argument
61 MS_EXCEPTION_IF_NULL(op_info); in SetKernelAttrs()
62 auto inputs_ptr = op_info->inputs_ptr(); in SetKernelAttrs()
63 auto outputs_ptr = op_info->outputs_ptr(); in SetKernelAttrs()
65 MS_LOG(EXCEPTION) << "op " << op_info->op_name() << " input size is zero."; in SetKernelAttrs()
83 if (same_op_name.count(op_info->op_name()) != 0) { in SetKernelAttrs()
/third_party/mindspore/mindspore/profiler/parser/
Dminddata_pipeline_parser.py155 op_info = pipeline_info.get('op_info')
156 if sample_interval is None or not op_info:
162 for item in op_info:
274 op_info = [
286 return op_info
/third_party/mindspore/mindspore/ops/_op_impl/akg/ascend/
Dbatchmatmul.py19 op_info = AkgAscendRegOp("BatchMatMul") \ variable
30 @op_info_register(op_info)
Dprod_force_se_a.py19 op_info = AkgAscendRegOp("ProdForceSeA") \ variable
30 @op_info_register(op_info)
Dload_im2col.py19 op_info = AkgAscendRegOp("LoadIm2Col") \ variable
30 @op_info_register(op_info)
Dreduce_min.py19 op_info = AkgAscendRegOp("ReduceMin") \ variable
29 @op_info_register(op_info)
Dreduce_max.py19 op_info = AkgAscendRegOp("ReduceMax") \ variable
29 @op_info_register(op_info)
Dless.py19 op_info = AkgAscendRegOp("Less") \ variable
28 @op_info_register(op_info)
Dexpand_dims.py19 op_info = AkgAscendRegOp("ExpandDims") \ variable
30 @op_info_register(op_info)
Dreciprocal.py19 op_info = AkgAscendRegOp("Reciprocal") \ variable
29 @op_info_register(op_info)
/third_party/mindspore/mindspore/ops/
Dop_info_register.py31 def op_info_register(op_info): argument
69 if isinstance(op_info, dict):
70 op_info_real = json.dumps(op_info)
72 op_info_real = op_info
230 op_info = {}
233 op_info[key.rstrip('_')] = value
235 op_info[key] = value
236 return op_info

1234