From 0a878ae1bbb13002e50f8287721750d2e4b22680 Mon Sep 17 00:00:00 2001 From: Jaroslaw Rzepecki Date: Wed, 22 Nov 2017 17:16:39 +0000 Subject: COMPMID-556: Added a rounding policy to the quantize function Change-Id: I6272a36636c5d9baff6d35dee0a50dc847f65bfa Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/110266 Tested-by: BSG Visual Compute Jenkins server to access repositories on http://mpd-gerrit.cambridge.arm.com Reviewed-by: Anthony Barbier --- src/core/Utils.cpp | 34 ++++++++++++++++++++++++++++++++++ 1 file changed, 34 insertions(+) (limited to 'src/core/Utils.cpp') diff --git a/src/core/Utils.cpp b/src/core/Utils.cpp index af864f57f7..af50bbbaf7 100644 --- a/src/core/Utils.cpp +++ b/src/core/Utils.cpp @@ -21,10 +21,13 @@ * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE * SOFTWARE. */ + #include "arm_compute/core/Utils.h" #include "arm_compute/core/FixedPoint.h" +#include "support/ToolchainSupport.h" + #include #include #include @@ -387,3 +390,34 @@ int arm_compute::max_consecutive_elements_display_width(std::ostream &s, DataTyp } return 0; } + +int arm_compute::round(float x, RoundingPolicy rounding_policy) +{ + using namespace std; + int rounded = 0; + switch(rounding_policy) + { + case RoundingPolicy::TO_ZERO: + { + rounded = static_cast(x); + break; + } + case RoundingPolicy::TO_NEAREST_UP: + { + rounded = static_cast(support::cpp11::round(x)); + break; + } + case RoundingPolicy::TO_NEAREST_EVEN: + { + ARM_COMPUTE_ERROR("TO_NEAREST_EVEN rounding policy is not supported."); + break; + } + default: + { + ARM_COMPUTE_ERROR("Unsupported rounding policy."); + break; + } + } + + return rounded; +} -- cgit v1.2.1