From 8c5e3dc9f04ac15aa926d1a1214610760830dbe2 Mon Sep 17 00:00:00 2001
From: Francis Murtagh <francis.murtagh@arm.com>
Date: Thu, 30 Aug 2018 17:18:37 +0100
Subject: IVGCVSW-1786 - Division by zero

 * Added logic for different cases of division by zero for ref
 * Added DivisionByZeroTest
 * Updated SelectiveComparer to compare NAN and INFINITY

Change-Id: Iec6f89264b17a0b03fad5d7ec4b2dafc31fea5df
---
 src/armnn/backends/RefWorkloads/Division.cpp | 41 ++++++++++++++++++++++++++--
 1 file changed, 39 insertions(+), 2 deletions(-)

(limited to 'src/armnn/backends/RefWorkloads')

diff --git a/src/armnn/backends/RefWorkloads/Division.cpp b/src/armnn/backends/RefWorkloads/Division.cpp
index 9837fea6b4..9c1142226c 100644
--- a/src/armnn/backends/RefWorkloads/Division.cpp
+++ b/src/armnn/backends/RefWorkloads/Division.cpp
@@ -8,6 +8,8 @@
 
 #include <functional>
 
+#include <cmath>
+
 namespace
 {
 
@@ -18,8 +20,43 @@ void ElementwiseDivision(unsigned int numElements,
 {
     for (unsigned int i = 0; i < numElements; ++i)
     {
-        //TODO How to handle divide by 0
-        outData[i] = inData0[i] / inData1[i];
+        if (inData1[i] != 0.0f)
+        {
+            outData[i] = inData0[i] / inData1[i];
+        }
+        else if (inData0[i] == 0.0f)
+        {
+            if (!std::signbit(inData1[i]))
+            {
+                outData[i]= NAN;
+            }
+            else
+            {
+                outData[i]= -NAN;
+            }
+        }
+        else if (inData0[i] < 0.0f)
+        {
+            if (!std::signbit(inData1[i]))
+            {
+                outData[i] = -INFINITY;
+            }
+            else
+            {
+                outData[i] = INFINITY;
+            }
+        }
+        else
+        {
+            if (!std::signbit(inData1[i]))
+            {
+                outData[i] = INFINITY;
+            }
+            else
+            {
+                outData[i] = -INFINITY;
+            }
+        }
     }
 }
 
-- 
cgit v1.2.1