Skip to content

Commit 6bbbc38

Browse files
committed
Fix formatting issues in SQ8 inner product function and clean up conditional compilation in tests
1 parent e50dc45 commit 6bbbc38

File tree

2 files changed

+13
-14
lines changed

2 files changed

+13
-14
lines changed

src/VecSim/spaces/IP/IP_AVX512F_BW_VL_VNNI_SQ8.h

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -134,7 +134,7 @@ template <unsigned char residual> // 0..15
134134
float SQ8_InnerProductSIMD16_AVX512F_BW_VL_VNNI(const void *pVec1v, const void *pVec2v,
135135
size_t dimension) {
136136
// The inner product similarity is 1 - ip
137-
return 1.0f -SQ8_InnerProductImp_AVX512<residual>(pVec1v, pVec2v, dimension);;
137+
return 1.0f - SQ8_InnerProductImp_AVX512<residual>(pVec1v, pVec2v, dimension);
138138
}
139139

140140
template <unsigned char residual> // 0..15

tests/unit/test_spaces.cpp

Lines changed: 12 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -308,7 +308,6 @@ TEST_F(SpacesTest, uint8_Cosine_no_optimization_func_test) {
308308
ASSERT_NEAR(dist, 0.0, 0.000001);
309309
}
310310

311-
312311
void common_ip_sq8(bool should_normalize, float expected_dist) {
313312

314313
size_t dim = 5;
@@ -367,14 +366,14 @@ void common_ip_sq8(bool should_normalize, float expected_dist) {
367366
<< "SQ8_InnerProduct failed to match expected distance";
368367

369368
unsigned char alignment = 0;
370-
#ifdef CPU_FEATURES_ARCH_AARCH64
369+
#ifdef CPU_FEATURES_ARCH_AARCH64
371370
// Make sure we don't use any optimization (because there is no size optimization for arm)
372371
auto optimization = getCpuOptimizationFeatures();
373372
optimization.sve = optimization.sve2 = optimization.asimddp = optimization.asimd = 0;
374373
auto arch_opt_func = IP_SQ8_GetDistFunc(dim, &alignment, &optimization);
375-
#else
374+
#else
376375
auto arch_opt_func = IP_SQ8_GetDistFunc(dim, &alignment, nullptr);
377-
#endif
376+
#endif
378377
ASSERT_EQ(arch_opt_func, SQ8_InnerProduct)
379378
<< "Unexpected distance function chosen for dim " << dim;
380379
ASSERT_NEAR(baseline, arch_opt_func(v1_orig, v2_quantized.data(), dim), 0.01)
@@ -405,14 +404,14 @@ TEST_F(SpacesTest, SQ8_Cosine_no_optimization_func_test) {
405404
float baseline = SQ8_Cosine(v1_orig.data(), v2_quantized.data(), dim);
406405

407406
unsigned char alignment = 0;
408-
#ifdef CPU_FEATURES_ARCH_AARCH64
407+
#ifdef CPU_FEATURES_ARCH_AARCH64
409408
// Make sure we don't use any optimization (because there is no size optimization for arm)
410409
auto optimization = getCpuOptimizationFeatures();
411410
optimization.sve = optimization.sve2 = optimization.asimddp = optimization.asimd = 0;
412411
auto arch_opt_func = Cosine_SQ8_GetDistFunc(dim, &alignment, &optimization);
413-
#else
412+
#else
414413
auto arch_opt_func = Cosine_SQ8_GetDistFunc(dim, &alignment, nullptr);
415-
#endif
414+
#endif
416415
ASSERT_EQ(arch_opt_func, SQ8_Cosine) << "Unexpected distance function chosen for dim " << dim;
417416
ASSERT_NEAR(baseline, arch_opt_func(v1_orig.data(), v2_quantized.data(), dim), 0.01)
418417
<< "No optimization with dim " << dim;
@@ -2356,14 +2355,14 @@ TEST_F(SpacesTest, SQ8_SQ8_ip_no_optimization_func_test) {
23562355
v2_quantized.data(), dim);
23572356

23582357
unsigned char alignment = 0;
2359-
#ifdef CPU_FEATURES_ARCH_AARCH64
2358+
#ifdef CPU_FEATURES_ARCH_AARCH64
23602359
// Make sure we don't use any optimization (because there is no size optimization for arm)
23612360
auto optimization = getCpuOptimizationFeatures();
23622361
optimization.sve = optimization.sve2 = optimization.asimddp = optimization.asimd = 0;
23632362
auto arch_opt_func = IP_SQ8_SQ8_GetDistFunc(dim, &alignment, &optimization);
2364-
#else
2363+
#else
23652364
auto arch_opt_func = IP_SQ8_SQ8_GetDistFunc(dim, &alignment, nullptr);
2366-
#endif
2365+
#endif
23672366
ASSERT_EQ(arch_opt_func, SQ8_SQ8_InnerProduct)
23682367
<< "Unexpected distance function chosen for dim " << dim;
23692368
// Checks that the function with the optimzied math equivalence returns the same result.
@@ -2386,14 +2385,14 @@ TEST_F(SpacesTest, SQ8_SQ8_Cosine_no_optimization_func_test) {
23862385
test_utils::SQ8_SQ8_NotOptimized_Cosine(v1_quantized.data(), v2_quantized.data(), dim);
23872386

23882387
unsigned char alignment = 0;
2389-
#ifdef CPU_FEATURES_ARCH_AARCH64
2388+
#ifdef CPU_FEATURES_ARCH_AARCH64
23902389
// Make sure we don't use any optimization (because there is no size optimization for arm)
23912390
auto optimization = getCpuOptimizationFeatures();
23922391
optimization.sve = optimization.sve2 = optimization.asimddp = optimization.asimd = 0;
23932392
auto arch_opt_func = Cosine_SQ8_SQ8_GetDistFunc(dim, &alignment, &optimization);
2394-
#else
2393+
#else
23952394
auto arch_opt_func = Cosine_SQ8_SQ8_GetDistFunc(dim, &alignment, nullptr);
2396-
#endif
2395+
#endif
23972396
ASSERT_EQ(arch_opt_func, SQ8_SQ8_Cosine)
23982397
<< "Unexpected distance function chosen for dim " << dim;
23992398
// Checks that the function with the optimzied math equivalence returns the same result.

0 commit comments

Comments
 (0)