Assert that attribute can be down-casted to tensor attribute.

1 files changed, 13 insertions, 1 deletions

diff --git a/searchlib/src/vespa/searchlib/tensor/imported_tensor_attribute_vector_read_guard.cpp b/searchlib/src/vespa/searchlib/tensor/imported_tensor_attribute_vector_read_guard.cpp
index 9ea598d3578..4f355255db5 100644
--- a/searchlib/src/vespa/searchlib/tensor/imported_tensor_attribute_vector_read_guard.cpp
+++ b/searchlib/src/vespa/searchlib/tensor/imported_tensor_attribute_vector_read_guard.cpp
@@ -7,11 +7,23 @@ namespace search::tensor {
 
 using vespalib::tensor::Tensor;
 
+namespace {
+
+const ITensorAttribute &
+getTensorAttribute(const search::attribute::IAttributeVector &attr)
+{
+    const ITensorAttribute *result = attr.asTensorAttribute();
+    assert(result != nullptr);
+    return *result;
+}
+
+}
+
 ImportedTensorAttributeVectorReadGuard::ImportedTensorAttributeVectorReadGuard(const attribute::ImportedAttributeVector &imported_attribute,
                                                                                bool stableEnumGuard)
     : ImportedAttributeVectorReadGuard(imported_attribute,
                                        stableEnumGuard),
-      _target_tensor_attribute(*imported_attribute.getTargetAttribute()->asTensorAttribute())
+      _target_tensor_attribute(getTensorAttribute(*imported_attribute.getTargetAttribute()))
 {
 }