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