add key_caching in pull

dmlc · Apr 10, 2015 · 0d1995c · 0d1995c
1 parent 30ffda6
commit 0d1995c
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/app/linear_method/async_sgd.h b/src/app/linear_method/async_sgd.h
@@ -228,6 +228,7 @@ class AsyncSGDWorker : public ISGDCompNode, public LinearMethod {
  MessagePtr msg(new Message(kServerGroup));
  msg->setKey(key);
  msg->task.set_key_channel(id);
+ msg->addFilter(FilterConfig::KEY_CACHING)
  msg->fin_handle = [this, id]() { computeGradient(id); };
  model_.key(id) = key;
  model_.pull(msg);