[coco] Support Op destruction (#1369)

This commit introduces destroy method into OpManager class, which allows
users to destruct Op instances.

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

diff --git a/contrib/coco/core/include/coco/IR/OpManager.h b/contrib/coco/core/include/coco/IR/OpManager.h
index 376a202..05e4f98 100644 (file)
--- a/contrib/coco/core/include/coco/IR/OpManager.h
+++ b/contrib/coco/core/include/coco/IR/OpManager.h
@@ -25,6 +25,13 @@ public:
 public:
   template <typename T> T *create(void);
 
+public:
+  // @brief Destroy (= deallocate) a Op instance
+  //
+  // NOTE OpManager will invoke dispose before deallocation
+  // NOTE destroy(op) WILL NOT update op->parent(). Client SHOULD detach op before destroy(op) call
+  void destroy(Op *);
+
 private:
   const PtrLink<Op, Instr> *const _op_link;
   const PtrLink<Object, ObjectInfo> *const _obj_link;

diff --git a/contrib/coco/core/src/IR/OpManager.cpp b/contrib/coco/core/src/IR/OpManager.cpp
index 8d09206..d17627a 100644 (file)
--- a/contrib/coco/core/src/IR/OpManager.cpp
+++ b/contrib/coco/core/src/IR/OpManager.cpp
@@ -2,6 +2,8 @@
 
 #include <nncc/foundation/Memory.h>
 
+#include <cassert>
+
 using nncc::foundation::make_unique;
 
 namespace coco
@@ -30,4 +32,11 @@ template <> AvgPool2D *OpManager::create<AvgPool2D>(void)
 
 template <> ReLU *OpManager::create<ReLU>(void) { return take(make_unique<ReLU>(_op_link)); }
 
+void OpManager::destroy(Op *op)
+{
+  assert(op->parent() == nullptr);
+  op->dispose();
+  release(op);
+}
+
 } // namespace coco

diff --git a/contrib/coco/core/src/IR/OpManager.test.cpp b/contrib/coco/core/src/IR/OpManager.test.cpp
index 9f02f0e..908ea6d 100644 (file)
--- a/contrib/coco/core/src/IR/OpManager.test.cpp
+++ b/contrib/coco/core/src/IR/OpManager.test.cpp
@@ -43,3 +43,10 @@ TEST_F(OpManagerTest, ReLU)
 
   ASSERT_NE(obj, nullptr);
 }
+
+TEST_F(OpManagerTest, destroy)
+{
+  auto op = mgr.create<coco::Conv2D>();
+  mgr.destroy(op);
+  ASSERT_EQ(mgr.size(), 0);
+}