Return NULL Index if SVS LVQ is requested but not supported.

Author: rfsaliev

src/VecSim/algorithms/svs/svs_utils.h

@@ -6,6 +6,11 @@
 #include "svs/lib/float16.h"
 #include "svs/index/vamana/dynamic_index.h"
 
+#include <cpuid.h>
+#include <cstdint>
+#include <cstdlib>
+#include <string>
+
 namespace svs_details {
 // VecSim->SVS data type conversion
 template <typename T>
@@ -133,6 +138,23 @@ inline svs::lib::PowerOfTwo SVSBlockSize(size_t bs, size_t elem_size) {
     return svs_bs;
 }
 
+bool check_cpuid() {
+    uint32_t eax, ebx, ecx, edx;
+    __cpuid(0, eax, ebx, ecx, edx);
+    std::string vendor_id = std::string((const char*)&ebx, 4) +
+                            std::string((const char*)&edx, 4) +
+                            std::string((const char*)&ecx, 4);
+    return (vendor_id == "GenuineIntel");
+}
+
+bool isSVSLVQModeSupported(VecSimSvsQuantBits quant_bits) {
+#if HAVE_SVS_LVQ
+    // Check if the CPU supports SVS LVQ
+    return check_cpuid();
+#endif
+    return quant_bits == VecSimSvsQuant_NONE;
+}
+
 } // namespace svs_details
 
 template <typename DataType, size_t QuantBits, size_t ResidualBits, class Enable = void>

src/VecSim/index_factories/svs_factory.cpp

@@ -35,6 +35,10 @@ VecSimIndex *NewIndexImpl(const VecSimParams *params, bool is_normalized) {
 
 template <typename MetricType, typename DataType>
 VecSimIndex *NewIndexImpl(const VecSimParams *params, bool is_normalized) {
+    if (!svs_details::isSVSLVQModeSupported(params->algoParams.svsParams.quantBits)) {
+        return NULL;
+    }
+
     switch (params->algoParams.svsParams.quantBits) {
     case VecSimSvsQuant_NONE:
         return NewIndexImpl<MetricType, DataType, 0>(params, is_normalized);