Add attribute access hint for tensorFromWeightedSet feature.

1 files changed, 3 insertions, 0 deletions

diff --git a/searchlib/src/vespa/searchlib/features/tensor_from_weighted_set_feature.cpp b/searchlib/src/vespa/searchlib/features/tensor_from_weighted_set_feature.cpp
index cbe262a0cbd..00eca7b680e 100644
--- a/searchlib/src/vespa/searchlib/features/tensor_from_weighted_set_feature.cpp
+++ b/searchlib/src/vespa/searchlib/features/tensor_from_weighted_set_feature.cpp
@@ -62,6 +62,9 @@ TensorFromWeightedSetBlueprint::setup(const search::fef::IIndexEnvironment &env,
     describeOutput("tensor",
                    "The tensor created from the given weighted set source (attribute field or query parameter)",
                    FeatureType::object(ValueType::make_type(CellType::DOUBLE, {{_dimension}})));
+    if (validSource && _sourceType == ATTRIBUTE_SOURCE) {
+        env.hintAttributeAccess(_sourceParam);
+    }
     return validSource;
 }