Browse Source

FormatAgnostic without checking format in kernel build info

tags/v0.5.0-beta
jjfeing 5 years ago
parent
commit
8278db9d9c
1 changed files with 6 additions and 1 deletions
  1. +6
    -1
      mindspore/ccsrc/kernel/kernel_build_info.cc

+ 6
- 1
mindspore/ccsrc/kernel/kernel_build_info.cc View File

@@ -105,7 +105,12 @@ bool KernelBuildInfo::operator==(const KernelBuildInfo &other) const {
return false;
}
if (inputs_format_ != other.inputs_format_ || outputs_format_ != other.outputs_format_) {
return false;
if (op_pattern_ != kFormatAgnosticPattern) {
return false;
} else {
MS_LOG(INFO) << "this kernel build info:" << this->ToString()
<< ", other kernel build info: " << other.ToString();
}
}
return !(inputs_device_type_ != other.inputs_device_type_ || outputs_device_type_ != other.outputs_device_type_);
}


Loading…
Cancel
Save