fix(quantization): remove incorrect query type conversion in ProcessQueryImpl

Query vectors are always FP32, do not reinterpret them as int8_t or uint16_t.
Also fix clang-format issues in test_fp16_bf16.cpp.

Signed-off-by: LHT129 <tianlan.lht@antgroup.com>

Author: LHT129

src/quantization/quantizer_adapter.cpp

@@ -221,31 +221,8 @@ template <typename QuantT, typename DataT>
 void
 QuantizerAdapter<QuantT, DataT>::ProcessQueryImpl(
     const DataType* query, Computer<QuantizerAdapter<QuantT, DataT>>& computer) const {
-    if constexpr (std::is_same_v<DataT, int8_t>) {
-        const auto* query_int8 = reinterpret_cast<const int8_t*>(query);
-        Vector<DataType> vec(this->dim_, this->allocator_);
-        for (int64_t i = 0; i < this->dim_; i++) {
-            vec[i] = static_cast<DataType>(query_int8[i]);
-        }
-        auto& inner_computer = reinterpret_cast<Computer<QuantT>&>(computer);
-        this->inner_quantizer_->ProcessQueryImpl(vec.data(), inner_computer);
-    } else if constexpr (std::is_same_v<DataT, uint16_t>) {
-        const auto* query_fp16 = reinterpret_cast<const uint16_t*>(query);
-        Vector<DataType> vec(this->dim_, this->allocator_);
-        for (int64_t i = 0; i < this->dim_; i++) {
-            if (data_type_ == DataTypes::DATA_TYPE_FP16) {
-                vec[i] = FP16ToFloat(query_fp16[i]);
-            } else {
-                vec[i] = BF16ToFloat(query_fp16[i]);
-            }
-        }
-        auto& inner_computer = reinterpret_cast<Computer<QuantT>&>(computer);
-        this->inner_quantizer_->ProcessQueryImpl(vec.data(), inner_computer);
-    } else {
-        static_assert(
-            std::is_same_v<DataT, int8_t> || std::is_same_v<DataT, uint16_t>,
-            "QuantizerAdapter::ProcessQueryImpl only supports int8_t and uint16_t data types");
-    }
+    auto& inner_computer = reinterpret_cast<Computer<QuantT>&>(computer);
+    this->inner_quantizer_->ProcessQueryImpl(query, inner_computer);
 }
 
 template <typename QuantT, typename DataT>

tests/test_fp16_bf16.cpp

@@ -35,11 +35,7 @@ CreateFP16Dataset(int64_t dim, int64_t count, const std::vector<float>& fp32_dat
     }
 
     auto dataset = Dataset::Make();
-    dataset->NumElements(count)
-        ->Dim(dim)
-        ->Ids(ids)
-        ->Float16Vectors(fp16_data)
-        ->Owner(true);
+    dataset->NumElements(count)->Dim(dim)->Ids(ids)->Float16Vectors(fp16_data)->Owner(true);
     return dataset;
 }
 
@@ -56,11 +52,7 @@ CreateBF16Dataset(int64_t dim, int64_t count, const std::vector<float>& fp32_dat
     }
 
     auto dataset = Dataset::Make();
-    dataset->NumElements(count)
-        ->Dim(dim)
-        ->Ids(ids)
-        ->Float16Vectors(bf16_data)
-        ->Owner(true);
+    dataset->NumElements(count)->Dim(dim)->Ids(ids)->Float16Vectors(bf16_data)->Owner(true);
     return dataset;
 }