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

1 files changed, 11 insertions, 0 deletions

diff --git a/src/armnn/test/TensorHelpers.hpp b/src/armnn/test/TensorHelpers.hpp
index ec38940a44..13ae1008cd 100644
--- a/src/armnn/test/TensorHelpers.hpp
+++ b/src/armnn/test/TensorHelpers.hpp
@@ -44,6 +44,17 @@ struct SelectiveComparer<T, false>
         {
             return std::abs(a - b) <= g_FloatCloseToZeroTolerance;
         }
+
+        if (std::isinf(a) && a == b)
+        {
+            return true;
+        }
+
+        if (std::isnan(a) && std::isnan(b))
+        {
+            return true;
+        }
+
         // For unquantized floats we use a tolerance of 1%.
         boost::math::fpc::close_at_tolerance<float> comparer(boost::math::fpc::percent_tolerance(1.0f));
         return comparer(a, b);