Commit 963a0419 by Siva Committed by Tianqi Chen

[NNVM] : Function header corrections. (#324)

parent 54051271
...@@ -121,7 +121,7 @@ class CompileEngine { ...@@ -121,7 +121,7 @@ class CompileEngine {
return GraphCacheEntry(); return GraphCacheEntry();
} }
} }
// Find the function given graph key. // Set the given function on given graph key.
void Set(const GraphKey& key, GraphFunc func) { void Set(const GraphKey& key, GraphFunc func) {
std::lock_guard<std::mutex> lock(mutex_); std::lock_guard<std::mutex> lock(mutex_);
std::shared_ptr<GraphCacheEntryNode> n = std::make_shared<GraphCacheEntryNode>(); std::shared_ptr<GraphCacheEntryNode> n = std::make_shared<GraphCacheEntryNode>();
...@@ -129,7 +129,7 @@ class CompileEngine { ...@@ -129,7 +129,7 @@ class CompileEngine {
n->use_count = 1; n->use_count = 1;
cache_[key] = GraphCacheEntry(n); cache_[key] = GraphCacheEntry(n);
} }
// Find the function given graph key. // Clear the function cache.
void Clear() { void Clear() {
std::lock_guard<std::mutex> lock(mutex_); std::lock_guard<std::mutex> lock(mutex_);
cache_.clear(); cache_.clear();
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment