[coco.generic] Introduce allocate for KernelObject (#1715)

This commit introduces allocate method in PlaingWeightContext as the
first step toward Bag-level data management.

Signed-off-by: Jonghyun Park <jh1302.park@samsung.com>

diff --git a/contrib/coco/generic/include/coco/IR/Data.h b/contrib/coco/generic/include/coco/IR/Data.h
index 38268a3..9a05ae6 100644 (file)
--- a/contrib/coco/generic/include/coco/IR/Data.h
+++ b/contrib/coco/generic/include/coco/IR/Data.h
@@ -17,6 +17,8 @@ struct Data
   virtual ~Data() = default;
 
   // @brief Allocate blob space of 'size' bytes
+  //
+  // WARN Deprecated
   virtual coco::BlobID allocate(uint32_t size) = 0;
 
   virtual PlainWeightContext<float> *f32(void) = 0;

diff --git a/contrib/coco/generic/include/coco/IR/PlainWeightContext.h b/contrib/coco/generic/include/coco/IR/PlainWeightContext.h
index ca6ad99..0354a26 100644 (file)
--- a/contrib/coco/generic/include/coco/IR/PlainWeightContext.h
+++ b/contrib/coco/generic/include/coco/IR/PlainWeightContext.h
@@ -19,6 +19,8 @@ template <typename T> struct PlainWeightContext
 {
   virtual ~PlainWeightContext() = default;
 
+  virtual void allocate(const KernelObject *) = 0;
+
   // @brief Link pre-allocated blob space (denoted by BlobID) with coco::KernelObject
   virtual void link(const BlobID &, const KernelObject *) = 0;
 

diff --git a/contrib/coco/generic/src/IR/Data.cpp b/contrib/coco/generic/src/IR/Data.cpp
index c10946a..c3b3671 100644 (file)
--- a/contrib/coco/generic/src/IR/Data.cpp
+++ b/contrib/coco/generic/src/IR/Data.cpp
@@ -49,6 +49,17 @@ public:
   PlainWeightContextImpl(PlainWeightContextImpl &&) = delete;
 
 public:
+  void allocate(const coco::KernelObject *obj)
+  {
+    assert(obj != nullptr);
+    auto bag = obj->bag();
+    assert(bag != nullptr);
+
+    auto id = _blob->allocate(sizeof(T) * bag->size());
+    link(id, obj);
+  }
+
+public:
   void link(const coco::BlobID &id, const coco::KernelObject *o)
   {
     _link[o] = reinterpret_cast<T *>(_blob->at(id));