| 281 | virtual ~BCDWorker() { } |
| 282 | |
| 283 | void process(const MessagePtr& msg) { |
| 284 | CHECK(msg->task.has_bcd()); |
| 285 | auto bcd = msg->task.bcd(); |
| 286 | int time = msg->task.time() * time_ratio_; |
| 287 | switch (bcd.cmd()) { |
| 288 | case BCDCall::LOAD_DATA: { |
| 289 | LoadDataReturn ret; |
| 290 | int hit_cache = 0; |
| 291 | CHECK(bcd.has_data()); |
| 292 | loadData(bcd.data(), ret.mutable_example_info(), &hit_cache); |
| 293 | ret.set_hit_cache(hit_cache); |
| 294 | sys_.replyProtocalMessage(msg, ret); |
| 295 | break; |
| 296 | } |
| 297 | case BCDCall::PREPROCESS_DATA: |
| 298 | preprocessData(time, bcd); |
| 299 | break; |
| 300 | case BCDCall::UPDATE_MODEL: |
| 301 | computeGradient(time, bcd, msg); |
| 302 | msg->finished = false; // |
| 303 | break; |
| 304 | case BCDCall::EVALUATE_PROGRESS: { |
| 305 | BCDProgress prog; evaluate(&prog); |
| 306 | sys_.replyProtocalMessage(msg, prog); |
| 307 | break; |
| 308 | } |
| 309 | default: break; |
| 310 | } |
| 311 | } |
| 312 | protected: |
| 313 | virtual void computeGradient(int time, const BCDCall& bcd, MessagePtr msg) = 0; |
| 314 | virtual void evaluate(BCDProgress* prog) = 0; |
nothing calls this directly
no test coverage detected