[unittests] Increase allowed error threshold while comparing floating

point tensors in BackendCorrectnessTest.
pytorch · Oct 12, 2018 · e0d3754 · e0d3754
1 parent 71c49e0
commit e0d3754
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tests/unittests/BackendCorrectnessTest.cpp b/tests/unittests/BackendCorrectnessTest.cpp
@@ -68,7 +68,7 @@ using CreateAndInitFunction =
 static bool
 compareAgainstInterpreter(BackendKind backendKind,
                           CreateAndInitFunction createAndInitFunction,
-                          bool quantize) {
+                          bool quantize, float allowedError = 0.0001) {
   ExecutionEngine IEE{BackendKind::Interpreter};
   ExecutionEngine BEE{backendKind};
   Context ICtx, BCtx;
@@ -95,7 +95,7 @@ compareAgainstInterpreter(BackendKind backendKind,
   IEE.run();
   BEE.run();
 
-  return IFT.second->isEqual(*BFT.second);
+  return IFT.second->isEqual(*BFT.second, allowedError);
 }
 
 } // namespace
@@ -496,7 +496,7 @@ TEST_P(BackendCorrectnessTest, convOps) {
 
 TEST_P(BackendCorrectnessTest, basicFCNet) {
   EXPECT_TRUE(compareAgainstInterpreter(GetParam(), createAndInitBasicFCNet,
-                                        /* quantize */ false));
+                                        /* quantize */ false, 0.0002));
 }
 
 TEST_P(BackendCorrectnessTest, basicFCNetQuantized) {