提交 3277a63e 编写于 作者: M mindspore-ci-bot 提交者: Gitee

!2227 modify create dynamic format build info

Merge pull request !2227 from jjfeing/master
...@@ -594,20 +594,9 @@ void TbeKernelSelect::CreateNewOpIOInfo(const mindspore::kernel::OpIOInfo &op_io ...@@ -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 // dtype && format
std::vector<std::string> dtype_new; op_io_info_new->set_dtypes(support_dtype);
for (size_t i = 0; i < support_format.size(); ++i) { op_io_info_new->set_formats(support_format);
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);
} }
void TbeKernelSelect::PrintSupportedFormat(const SupportFormat &support_format) { void TbeKernelSelect::PrintSupportedFormat(const SupportFormat &support_format) {
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册