diff --git a/mindspore/ccsrc/kernel/tbe/tbe_convert_utils.cc b/mindspore/ccsrc/kernel/tbe/tbe_convert_utils.cc index 3278021009..5212dc9641 100644 --- a/mindspore/ccsrc/kernel/tbe/tbe_convert_utils.cc +++ b/mindspore/ccsrc/kernel/tbe/tbe_convert_utils.cc @@ -74,14 +74,6 @@ TypeId DtypeToTypeId(const std::string &dtypes) { return iter->second; } -std::string DtypeToString(const std::string &dtypes) { - auto iter = type_str_maps.find(dtypes); - if (iter == type_str_maps.end()) { - MS_LOG(EXCEPTION) << "Illegal input dtype: " << dtypes; - } - return iter->second; -} - std::string TypeIdToString(TypeId type_id) { auto iter = type_id_str_maps.find(type_id); if (iter == type_id_str_maps.end()) { diff --git a/mindspore/ccsrc/kernel/tbe/tbe_convert_utils.h b/mindspore/ccsrc/kernel/tbe/tbe_convert_utils.h index 9b9b3770df..2c8d3008b9 100644 --- a/mindspore/ccsrc/kernel/tbe/tbe_convert_utils.h +++ b/mindspore/ccsrc/kernel/tbe/tbe_convert_utils.h @@ -28,8 +28,6 @@ namespace tbe { constexpr auto kProcessorAiCore = "aicore"; TypeId DtypeToTypeId(const std::string &dtypes); -std::string DtypeToString(const std::string &dtypes); - std::string TypeIdToString(TypeId type_id); size_t GetDtypeNbyte(const std::string &dtypes); diff --git a/mindspore/ccsrc/kernel/tbe/tbe_kernel_build.cc b/mindspore/ccsrc/kernel/tbe/tbe_kernel_build.cc index 86d6801a60..8a26de3400 100644 --- a/mindspore/ccsrc/kernel/tbe/tbe_kernel_build.cc +++ b/mindspore/ccsrc/kernel/tbe/tbe_kernel_build.cc @@ -615,7 +615,7 @@ void TbeKernelBuild::GenDescJson(const std::shared_ptr &anf_ spec_shape.emplace_back(shape[2] * shape[3]); spec_shape.emplace_back(shape[4]); (*output_desc)["shape"] = spec_shape; - } else if (fusion_data_type == kFusionReLUGradV2 && (*output_desc)["data_type"] == "uint8") { + } else if (fusion_data_type == kFusionReLUGradV2) { std::vector spec_shape = {}; spec_shape.emplace_back(shape[0]); spec_shape.emplace_back(shape[1]); diff --git a/mindspore/ccsrc/kernel/tbe/tbe_kernel_select/tbe_kernel_broadcast_selecter.h b/mindspore/ccsrc/kernel/tbe/tbe_kernel_select/tbe_kernel_broadcast_selecter.h index 2beb81927f..af711ddf29 100644 --- a/mindspore/ccsrc/kernel/tbe/tbe_kernel_select/tbe_kernel_broadcast_selecter.h +++ b/mindspore/ccsrc/kernel/tbe/tbe_kernel_select/tbe_kernel_broadcast_selecter.h @@ -51,7 +51,6 @@ class TbeKernelBroadCastSelecter { std::vector> input_shapes_; std::vector> output_shapes_; }; - } // namespace kernel } // namespace mindspore #endif // MINDSPORE_TBE_KERNEL_BROADCAST_SELECTER_HELPER_H diff --git a/mindspore/ccsrc/kernel/tbe/tbe_kernel_select/tbe_kernel_reduce_selecter.cc b/mindspore/ccsrc/kernel/tbe/tbe_kernel_select/tbe_kernel_reduce_selecter.cc index d62f3cea2f..da0466feaa 100644 --- a/mindspore/ccsrc/kernel/tbe/tbe_kernel_select/tbe_kernel_reduce_selecter.cc +++ b/mindspore/ccsrc/kernel/tbe/tbe_kernel_select/tbe_kernel_reduce_selecter.cc @@ -175,6 +175,5 @@ void TbeKernelReduceSelecter::PadScalarShape(std::vector *shape) const { shape->emplace_back(1); } } - } // namespace kernel } // namespace mindspore diff --git a/mindspore/ccsrc/kernel/tbe/tbe_kernel_select/tbe_kernel_select.cc b/mindspore/ccsrc/kernel/tbe/tbe_kernel_select/tbe_kernel_select.cc index e9ad128335..1b7ef2d72a 100644 --- a/mindspore/ccsrc/kernel/tbe/tbe_kernel_select/tbe_kernel_select.cc +++ b/mindspore/ccsrc/kernel/tbe/tbe_kernel_select/tbe_kernel_select.cc @@ -462,6 +462,9 @@ std::vector TbeKernelSelect::SplitStrToVec(const std::string &op_se std::vector ret; auto begin = op_select_tmp.find_first_not_of(space, 0); auto sep_pos = op_select_tmp.find(sep); + if (begin >= sep_pos) { + MS_LOG(EXCEPTION) << "Select ret json is error."; + } while (sep_pos != std::string::npos) { auto obj = op_select_tmp.substr(begin, sep_pos - begin); if (kDynamicFormatMap.find(obj) != kDynamicFormatMap.end()) { @@ -628,6 +631,5 @@ void TbeKernelSelect::PrintSupportedFormat(const SupportFormat &support_format) MS_LOG(INFO) << "Support format: " << print_str; } } - } // namespace kernel } // namespace mindspore