fix requantize flaky test (apache#16709)

yajiedesign · Nov 6, 2019 · 54faba0 · 54faba0
1 parent 044c7a1
commit 54faba0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/python/quantization/test_quantization.py b/tests/python/quantization/test_quantization.py
@@ -178,7 +178,7 @@ def check_requantize_with_symbol(shape, min_calib_range=None, max_calib_range=No
                                                                           max_range.asscalar(),
                                                                           min_calib_range=min_calib_range,
                                                                           max_calib_range=max_calib_range)
-        assert_almost_equal(qdata_int8.asnumpy(), qdata_int8_np)
+        assert_almost_equal(qdata_int8.asnumpy(), qdata_int8_np, atol = 1)
         assert_almost_equal(min_output.asnumpy(), np.array([min_output_np]))
         assert_almost_equal(max_output.asnumpy(), np.array([max_output_np]))