xiaoxue_admin
/
zvec
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix ut for quantized int (#47)

This commit is contained in:
ZeFeng Yin 2026-02-02 20:13:21 +08:00 committed by GitHub
parent 36e84f5e69
commit 6680c619be
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tests/core/metric/quantized_integer_metric_test.cc
View File

@ -192,7 +192,7 @@ TEST(QuantizedIntegerMetric, TestInt8SquaredEuclidean) {
float v2;
compute(mi, qi, holder2->dimension(), &v2);
// printf("%f %f\n", v1, v2);
ASSERT_NEAR(v1, v2, 1e-2 * DIMENSION);
ASSERT_NEAR(v1, v2, 1e-2 * (DIMENSION + 1));
std::string out2;
ASSERT_EQ(0, reformer->convert(iter->data(), qmeta, &out2, &qmeta2));
@ -516,7 +516,7 @@ void TestDistanceMatrixInt4(const std::string &metric_name) {
matrix_compute(&matrix2[0], &query2[0], meta2.dimension(), &result2[0]);
for (size_t i = 0; i < batch_size * query_size; ++i) {
EXPECT_FLOAT_EQ(result1[i], result2[i]);
EXPECT_NEAR(result1[i], result2[i], 1e-4);
EXPECT_TRUE(IsAlmostEqual(result1[i], result2[i], 1e4));
}
}