log info support dtype/format

pull/13052/head
jjfeing 4 years ago
parent 0c3152fe6b
commit c5292174f4

@ -35,13 +35,13 @@ kernel::KernelBuildInfoPtr GenerateKernelBuildInfo(const AnfNodePtr &concat) {
auto format_iter = std::find_if(inputs_device_format.begin(), inputs_device_format.end(),
[&](const auto &format) { return format != (*cmp_format); });
if (format_iter != inputs_device_format.end()) {
MS_LOG(EXCEPTION) << "Input format is not same, value: " << *format_iter;
MS_LOG(EXCEPTION) << "Input format is not same, value: " << (*format_iter) << ", need format: " << (*cmp_format);
}
auto cmp_dtype = inputs_device_type.begin();
auto dtype_iter = std::find_if(inputs_device_type.begin(), inputs_device_type.end(),
[&](const auto &dtype) { return dtype != (*cmp_dtype); });
if (dtype_iter != inputs_device_type.end()) {
MS_LOG(EXCEPTION) << "Input dtype is not same, value: " << *dtype_iter;
MS_LOG(EXCEPTION) << "Input dtype is not same, value: " << (*dtype_iter) << ", need dtype: " << (*cmp_dtype);
}
outputs_device_format.emplace_back(*cmp_format);
outputs_device_type.emplace_back(*cmp_dtype);

Loading…
Cancel
Save