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);