p04896132
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!2227 modify create dynamic format build info 

Merge pull request !2227 from jjfeing/master
This commit is contained in:
mindspore-ci-bot 2020-06-17 20:09:10 +08:00 committed by Gitee
parent ee27718176 838dd1e5dd
commit 3277a63e7d
1 changed files with 3 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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