From c8d6f7bb2b09b8928570222131353f75acb4007b Mon Sep 17 00:00:00 2001 From: Matthijs Douze Date: Wed, 6 Sep 2023 07:55:15 -0700 Subject: [PATCH] fix CI issues after cross-matrix diff (#3042) Summary: Pull Request resolved: https://github.com/facebookresearch/faiss/pull/3042 to fix nightly builds Reviewed By: mlomeli1 Differential Revision: D48969974 fbshipit-source-id: b7206aac907ed65caf182a95cf22ec463bb58dc4 --- faiss/MatrixStats.cpp | 2 +- tests/test_residual_quantizer.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/faiss/MatrixStats.cpp b/faiss/MatrixStats.cpp index 440d9f9dae..3eea7bf91f 100644 --- a/faiss/MatrixStats.cpp +++ b/faiss/MatrixStats.cpp @@ -12,7 +12,7 @@ #include /* va_list, va_start, va_arg, va_end */ #include -#include +#include #include #include diff --git a/tests/test_residual_quantizer.py b/tests/test_residual_quantizer.py index 5cc86f7161..2ca3e4c327 100644 --- a/tests/test_residual_quantizer.py +++ b/tests/test_residual_quantizer.py @@ -1117,7 +1117,7 @@ def test_precomp(self): K = 1 << rq.nbits.at(m) cpp_table = codebook_cross_prods[ofs:ofs + K * kk].reshape(kk, K) ofs += kk * K - np.testing.assert_allclose(py_table, cpp_table, rtol=1e-4) + np.testing.assert_allclose(py_table, cpp_table, rtol=2e-4) cent_norms = faiss.vector_to_array(rq.cent_norms) np.testing.assert_array_almost_equal(