Optimize NormalizeVec

Signed-off-by: Cai Yudong <[email protected]>
zilliztech · Aug 30, 2024 · 7764c6a · 7764c6a
1 parent 613f06f
commit 7764c6a
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/common/utils.cc b/src/common/utils.cc
@@ -41,7 +41,9 @@ NormalizeVec(DataType* x, int32_t d) {
     } else if constexpr (std::is_same_v<DataType, bf16>) {
         norm_l2_sqr = faiss::bf16_vec_norm_L2sqr(x, d);
     } else {
-        LOG_KNOWHERE_ERROR_ << "unknown datatype";
+        for (int32_t i = 0; i < d; i++) {
+            norm_l2_sqr += (float)x[i] * (float)x[i];
+        }
     }
 
     if (norm_l2_sqr > 0 && std::abs(1.0f - norm_l2_sqr) > FloatAccuracy) {