Merge pull request #26054 from vespa-engine/toregge/add-exponential-array-size-growth-to-tensor-buffer-type-mapperv8.127.1v8.126.20

Add exponential array size growth to tensor buffer type mapper.

1 files changed, 2 insertions, 1 deletions

diff --git a/searchlib/src/vespa/searchlib/tensor/tensor_buffer_type_mapper.h b/searchlib/src/vespa/searchlib/tensor/tensor_buffer_type_mapper.h
index 950076c9924..ad2116a429c 100644
--- a/searchlib/src/vespa/searchlib/tensor/tensor_buffer_type_mapper.h
+++ b/searchlib/src/vespa/searchlib/tensor/tensor_buffer_type_mapper.h
@@ -25,12 +25,13 @@ public:
     using LargeBufferType = LargeSubspacesBufferType;
 
     TensorBufferTypeMapper();
-    TensorBufferTypeMapper(uint32_t max_small_subspaces_type_id, TensorBufferOperations* ops);
+    TensorBufferTypeMapper(uint32_t max_small_subspaces_type_id, double grow_factor, TensorBufferOperations* ops);
     ~TensorBufferTypeMapper();
 
     uint32_t get_type_id(size_t array_size) const;
     size_t get_array_size(uint32_t type_id) const;
     TensorBufferOperations& get_tensor_buffer_operations() const noexcept { return *_ops; }
+    uint32_t get_max_small_array_type_id(uint32_t max_small_array_type_id) const noexcept;
 };
 
 }