Implement store for heap allocated tensors.

2 files changed, 8 insertions, 1 deletions

diff --git a/vespalib/src/vespa/vespalib/datastore/datastore.h b/vespalib/src/vespa/vespalib/datastore/datastore.h
index 193869a5591..4b908452b32 100644
--- a/vespalib/src/vespa/vespalib/datastore/datastore.h
+++ b/vespalib/src/vespa/vespalib/datastore/datastore.h
@@ -106,6 +106,7 @@ public:
     DataStore(const DataStore &rhs) = delete;
     DataStore &operator=(const DataStore &rhs) = delete;
     DataStore();
+    DataStore(uint32_t min_arrays);
     ~DataStore();
 
     EntryRef addEntry(const EntryType &e);
diff --git a/vespalib/src/vespa/vespalib/datastore/datastore.hpp b/vespalib/src/vespa/vespalib/datastore/datastore.hpp
index 6549425b022..ad35f3c7383 100644
--- a/vespalib/src/vespa/vespalib/datastore/datastore.hpp
+++ b/vespalib/src/vespa/vespalib/datastore/datastore.hpp
@@ -133,8 +133,14 @@ DataStoreT<RefT>::freeListRawAllocator(uint32_t typeId)
 
 template <typename EntryType, typename RefT>
 DataStore<EntryType, RefT>::DataStore()
+    : DataStore(RefType::offsetSize())
+{
+}
+
+template <typename EntryType, typename RefT>
+DataStore<EntryType, RefT>::DataStore(uint32_t min_arrays)
     : ParentType(),
-      _type(1, RefType::offsetSize(), RefType::offsetSize())
+      _type(1, min_arrays, RefType::offsetSize())
 {
     addType(&_type);
     initActiveBuffers();