updated object pool with upper size

tsung-wei-huang · tsung-wei-huang · commit a6e0eb8e6ec4 · 2019-07-11T23:32:52.000-05:00
diff --git a/benchmark/regression.py b/benchmark/regression.py
@@ -129,7 +129,7 @@ def main():
 
   plot.tight_layout()
   plot.savefig('result.png')
-  plot.show()
+  #plot.show()
   plot.close(fig)
 
 # run the main entry
diff --git a/taskflow/core/executor.hpp b/taskflow/core/executor.hpp
@@ -609,7 +609,7 @@ inline void Executor::_init_module_node(Node* node) {
   };
 }
 
-// Procedure: 
+// Procedure: _invoke
 inline void Executor::_invoke(unsigned me, Node* node) {
 
   // Here we need to fetch the num_successors first to avoid the invalid memory
diff --git a/taskflow/utility/object_pool.hpp b/taskflow/utility/object_pool.hpp
@@ -74,8 +74,10 @@ std::unique_ptr<T> ObjectPool<T>::destack(ArgsT&&... args) {
 // Procedure: enstack
 template <typename T>
 void ObjectPool<T>::enstack(std::unique_ptr<T>&& obj) {
-  obj->recycle();
-  _stack.push_back(std::move(obj));
+  if(_stack.size() < 4096) {
+    obj->recycle();
+    _stack.push_back(std::move(obj));
+  }
 }
 
 }  // end of namespace tf -----------------------------------------------------

Original file line number	Diff line number	Diff line change
`@@ -609,7 +609,7 @@ inline void Executor::_init_module_node(Node* node) {`
`609`	`609`	`};`
`610`	`610`	`}`
`611`	`611`
`612`		`-// Procedure:`
	`612`	`+// Procedure: _invoke`
`613`	`613`	`inline void Executor::_invoke(unsigned me, Node* node) {`
`614`	`614`
`615`	`615`	`// Here we need to fetch the num_successors first to avoid the invalid memory`