Skip to content
Snippets Groups Projects
Commit 07b3797c authored by Benjamin Halimi's avatar Benjamin Halimi
Browse files

use INFINITY instead of (inputValue + 1)

parent 9fd5d0be
No related branches found
No related tags found
1 merge request!168Draft: Feat : Add halfway rounding modes to the Round operator
Pipeline #71704 passed
...@@ -41,7 +41,7 @@ void RoundImpl_cpu_forward_kernel( ...@@ -41,7 +41,7 @@ void RoundImpl_cpu_forward_kernel(
output[i] = static_cast<O>(std::round(inputValue)); output[i] = static_cast<O>(std::round(inputValue));
} }
else if (mode == Round_Op::HalfwayRounding::NextInteger) { else if (mode == Round_Op::HalfwayRounding::NextInteger) {
output[i] = static_cast<O>(std::round(std::nextafter(inputValue, inputValue + 1))); output[i] = static_cast<O>(std::round(std::nextafter(inputValue, +INFINITY)));
} }
} }
} }
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment