| @@ -594,20 +594,9 @@ void TbeKernelSelect::CreateNewOpIOInfo(const mindspore::kernel::OpIOInfo &op_io | |||||
| op_io_info_new->set_need_compile(op_io_info.need_compile()); | op_io_info_new->set_need_compile(op_io_info.need_compile()); | ||||
| op_io_info_new->set_reshape_type(op_io_info.reshape_type()); | op_io_info_new->set_reshape_type(op_io_info.reshape_type()); | ||||
| op_io_info_new->set_shape(op_io_info.shape()); | op_io_info_new->set_shape(op_io_info.shape()); | ||||
| // dtype | |||||
| std::vector<std::string> dtype_new; | |||||
| for (size_t i = 0; i < support_format.size(); ++i) { | |||||
| dtype_new.insert(dtype_new.end(), support_dtype.begin(), support_dtype.end()); | |||||
| } | |||||
| op_io_info_new->set_dtypes(dtype_new); | |||||
| // format | |||||
| std::vector<std::string> format_new; | |||||
| for (const auto &format : support_format) { | |||||
| for (size_t j = 0; j < support_dtype.size(); ++j) { | |||||
| format_new.emplace_back(format); | |||||
| } | |||||
| } | |||||
| op_io_info_new->set_formats(format_new); | |||||
| // dtype && format | |||||
| op_io_info_new->set_dtypes(support_dtype); | |||||
| op_io_info_new->set_formats(support_format); | |||||
| } | } | ||||
| void TbeKernelSelect::PrintSupportedFormat(const SupportFormat &support_format) { | void TbeKernelSelect::PrintSupportedFormat(const SupportFormat &support_format) { | ||||