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

!49884 Fix diag op floating point exception, core dump. 

Merge pull request !49884 from xiaohanzhang/fix_diag
This commit is contained in:
i-robot 2023-03-07 07:33:09 +00:00 committed by Gitee
parent 04a3f8244d 02239dc20c
commit 8e0fbb61c5
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
mindspore/ccsrc/plugin/device/gpu/kernel/arrays/diag_gpu_kernel.cc
View File

@ -110,6 +110,10 @@ int DiagGpuKernelMod::Resize(const BaseOperatorPtr &base_operator, const std::ve
for (size_t i = LongToSize(batch_rank_); i < input_shape.size(); ++i) {
input_size_ *= LongToSize(input_shape[i]);
}
if (input_size_ == 0) {
MS_LOG(ERROR) << kernel_name_ << "input size should should be larger than 0, but got: " << input_size_;
return KRET_RESIZE_FAILED;
}
// Get the output size of each batch.
auto output = outputs.at(kIndex0);