Fixed complex version of the TRSV kernel

1 files changed, 4 insertions, 0 deletions

diff --git a/test/correctness/tester.cpp b/test/correctness/tester.cpp
index c449b09d..dc0f842e 100644
--- a/test/correctness/tester.cpp
+++ b/test/correctness/tester.cpp
@@ -410,6 +410,10 @@ bool TestSimilarityNear(const T val1, const T val2,
   if (val1 == val2) {
     return true;
   }
+  // Handles cases with both results NaN
+  else if (std::isnan(val1) && std::isnan(val2)) {
+    return true;
+  }
   // The values are zero or very small: the relative error is less meaningful
   else if (val1 == 0 || val2 == 0 || difference < error_margin_absolute) {
     return (difference < error_margin_absolute);