Browse Source

modify create dynamic format build info

tags/v0.5.0-beta
jjfeing 5 years ago
parent
commit
838dd1e5dd
1 changed files with 3 additions and 14 deletions
  1. +3
    -14
      mindspore/ccsrc/kernel/tbe/tbe_kernel_select/tbe_kernel_select.cc

+ 3
- 14
mindspore/ccsrc/kernel/tbe/tbe_kernel_select/tbe_kernel_select.cc View File

@@ -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<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) {


Loading…
Cancel
Save