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

!40339 fix range accuracy and add exception information 

Merge pull request !40339 from NaCN/fix_range_float
This commit is contained in:
i-robot 2022-08-15 07:17:41 +00:00 committed by Gitee
parent b2ad845781 9403157b29
commit 6ce47202fe
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
1 changed files with 10 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
mindspore/ccsrc/plugin/device/cpu/kernel/range_cpu_kernel.cc
View File

@ -82,14 +82,20 @@ void RangeCpuKernelMod::LaunchKernel(const std::vector<AddressPtr> &inputs, cons
auto output = reinterpret_cast<T *>(outputs[0]->addr);
size_t max_size = outputs[0]->size / sizeof(T);
if (Sign(delta) * Sign(limit - start) > 0) {
output_size_ = static_cast<size_t>(std::ceil(static_cast<double>(limit - start) / static_cast<double>(delta)));
if (Sign(delta) * Sign(limit - start) >= 0) {
if (dtype_ == kNumberTypeInt32 || dtype_ == kNumberTypeInt64) {
output_size_ = static_cast<size_t>((limit - start + std::abs(delta) - 1) / delta);
} else {
output_size_ = static_cast<size_t>(std::ceil((limit - start) / delta));
}
if (output_size_ > max_size) {
MS_LOG(EXCEPTION) << "For " << kernel_name_ << ", the output element number exceeds the maximum number.";
}
for (size_t index = 0; index < output_size_; index++, start += delta) {
output[index] = start;
for (size_t index = 0; index < output_size_; index++) {
output[index] = delta * index + start;
}
} else {
MS_LOG(EXCEPTION) << "For " << kernel_name_ << ", upper bound and larger bound inconsistent with step sign.";
}
}