From 838dd1e5dde391678b01bf4243bf1e214bf0fec1 Mon Sep 17 00:00:00 2001 From: jjfeing Date: Wed, 17 Jun 2020 16:32:54 +0800 Subject: [PATCH] modify create dynamic format build info --- .../tbe/tbe_kernel_select/tbe_kernel_select.cc | 17 +++-------------- 1 file changed, 3 insertions(+), 14 deletions(-) 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 1b7ef2d72a1..573ad176cfc 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 @@ -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_reshape_type(op_io_info.reshape_type()); op_io_info_new->set_shape(op_io_info.shape()); - // dtype - std::vector 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 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) {