Handle innerproduct distance metric when reading attribute header.

2 files changed, 4 insertions, 0 deletions

diff --git a/searchlib/src/tests/attribute/attribute_header/attribute_header_test.cpp b/searchlib/src/tests/attribute/attribute_header/attribute_header_test.cpp
index 1e89fbe7501..aca9b652b94 100644
--- a/searchlib/src/tests/attribute/attribute_header/attribute_header_test.cpp
+++ b/searchlib/src/tests/attribute/attribute_header/attribute_header_test.cpp
@@ -72,6 +72,8 @@ TEST(AttributeHeaderTest, can_be_added_to_and_extracted_from_generic_header)
     verify_roundtrip_serialization(HnswIPO({16, 100, DistanceMetric::Euclidean}));
     verify_roundtrip_serialization(HnswIPO({16, 100, DistanceMetric::Angular}));
     verify_roundtrip_serialization(HnswIPO({16, 100, DistanceMetric::GeoDegrees}));
+    verify_roundtrip_serialization(HnswIPO({16, 100, DistanceMetric::InnerProduct}));
+    verify_roundtrip_serialization(HnswIPO({16, 100, DistanceMetric::Hamming}));
     verify_roundtrip_serialization(HnswIPO());
 }
 
diff --git a/searchlib/src/vespa/searchlib/attribute/attribute_header.cpp b/searchlib/src/vespa/searchlib/attribute/attribute_header.cpp
index 430f2eaa560..cfbca71bd5e 100644
--- a/searchlib/src/vespa/searchlib/attribute/attribute_header.cpp
+++ b/searchlib/src/vespa/searchlib/attribute/attribute_header.cpp
@@ -114,6 +114,8 @@ to_distance_metric(const vespalib::string& metric)
         return DistanceMetric::Angular;
     } else if (metric == geodegrees) {
         return DistanceMetric::GeoDegrees;
+    } else if (metric == innerproduct) {
+        return DistanceMetric::InnerProduct;
     } else if (metric == hamming) {
         return DistanceMetric::Hamming;
     } else {