diff --git a/mindspore/ccsrc/backend/kernel_compiler/tbe/tbe_kernel_build.cc b/mindspore/ccsrc/backend/kernel_compiler/tbe/tbe_kernel_build.cc index 27343434e2..57aa4ffbd4 100644 --- a/mindspore/ccsrc/backend/kernel_compiler/tbe/tbe_kernel_build.cc +++ b/mindspore/ccsrc/backend/kernel_compiler/tbe/tbe_kernel_build.cc @@ -165,6 +165,10 @@ bool TbeKernelJsonCreator::GenTbeSingleKernelJson(const std::shared_ptrfullname_with_scope(); json_name_ = op_name + "_" + std::to_string(hash_id) + "_" + std::to_string(device_id); json_info_ = json_str; + auto iter = tbe::opTypeAdapter.find(op_name); + if (iter != tbe::opTypeAdapter.end()) { + op_name = iter->second; + } op_info_json["Type"] = op_name; op_info_json["graph_id"] = AnfAlgo::GetGraphId(anf_node.get()); op_info_json[kJKernelName] = json_name_; diff --git a/mindspore/ccsrc/backend/kernel_compiler/tbe/tbe_utils.h b/mindspore/ccsrc/backend/kernel_compiler/tbe/tbe_utils.h index e367e9d82e..58cd439eec 100644 --- a/mindspore/ccsrc/backend/kernel_compiler/tbe/tbe_utils.h +++ b/mindspore/ccsrc/backend/kernel_compiler/tbe/tbe_utils.h @@ -33,6 +33,8 @@ namespace kernel { namespace tbe { using std::string; using std::vector; +const std::map opTypeAdapter = { + {"ReLUV2", "ReluV2"}, {"ReLU6", "Relu6"}, {"ReLU6Grad", "Relu6Grad"}, {"ReLUGrad", "ReluGrad"}, {"ReLU", "Relu"}}; class TbeUtils { public: