xxq250
/
mindspore
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

!39150 fix scalar type compute of maximum and pclint issue. 

Merge pull request !39150 from yangshuo/br_pclint
This commit is contained in:
i-robot 2022-08-02 01:59:11 +00:00 committed by Gitee
parent a2f8f03a48 b3e0202ba8
commit 1b956ec349
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
2 changed files with 1 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mindspore/ccsrc/plugin/device/cpu/kernel/argmax_cpu_kernel.cc
View File

@ -78,7 +78,7 @@ bool ArgmaxCpuKernelMod::LaunchKernel(const std::vector<kernel::AddressPtr> &inp
}
bool ArgmaxCpuKernelMod::Init(const BaseOperatorPtr &base_operator, const std::vector<KernelTensorPtr> &inputs,
const std::vector<KernelTensorPtr> &outputs) {
const std::vector<KernelTensorPtr> &) {
auto kernel_ptr = std::dynamic_pointer_cast<ops::ArgMax>(base_operator);
if (!kernel_ptr) {
MS_LOG(ERROR) << "cast ArgMax ops failed!";

4
mindspore/ccsrc/plugin/device/cpu/kernel/maximum_cpu_kernel.cc
View File

@ -77,10 +77,6 @@ int MaximumCpuKernelMod::Resize(const BaseOperatorPtr &base_operator, const std:
InitInputTensorAndScalar(max_input_shape_size);
} else if (max_input_shape_size == output_shape_.size() && output_shape_.size() != 0) {
InitInputTensors(input_x_dtype, input_y_dtype);
} else {
MS_LOG(WARNING) << "For '" << kernel_name_ << "', inputs must be two tensors or one tensor and one scalar, but got "
<< input_x_dtype << " and " << input_y_dtype;
return KRET_RESIZE_FAILED;
}
return 0;
}