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

fix error log of kernel object types

This commit is contained in:
wYann 2023-02-10 10:30:53 +08:00
parent bbca90ef51
commit caa0db3bb0
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
mindspore/ccsrc/kernel/kernel_build_info.cc
View File

@ -86,6 +86,9 @@ TypeId KernelBuildInfo::GetOutputDeviceType(size_t output_index) const {
}
KernelObjectType KernelBuildInfo::GetInputKernelObjectType(size_t input_index) const {
if (inputs_kernel_object_type_.empty()) {
return KernelObjectType::UNKNOWN_TYPE;
}
if (input_index >= inputs_kernel_object_type_.size()) {
bool has_tuple_unfold =
std::any_of(inputs_kernel_object_type_.begin(), inputs_kernel_object_type_.end(),
@ -101,6 +104,9 @@ KernelObjectType KernelBuildInfo::GetInputKernelObjectType(size_t input_index) c
}
KernelObjectType KernelBuildInfo::GetOutputKernelObjectType(size_t output_index) const {
if (outputs_kernel_object_type_.empty()) {
return KernelObjectType::UNKNOWN_TYPE;
}
if (output_index >= outputs_kernel_object_type_.size()) {
bool has_tuple_unfold =
std::any_of(outputs_kernel_object_type_.begin(), outputs_kernel_object_type_.end(),