From f5245c615c9dcb96c72ec056b11fc12ea5afe0fe Mon Sep 17 00:00:00 2001 From: tqchen Date: Fri, 16 Jan 2015 10:12:47 -0800 Subject: [PATCH] ok --- rabit-learn/kmeans/.gitignore | 2 ++ src/allreduce_mock.h | 5 +++++ 2 files changed, 7 insertions(+) create mode 100644 rabit-learn/kmeans/.gitignore diff --git a/rabit-learn/kmeans/.gitignore b/rabit-learn/kmeans/.gitignore new file mode 100644 index 000000000..5f8241b66 --- /dev/null +++ b/rabit-learn/kmeans/.gitignore @@ -0,0 +1,2 @@ +kmeans +*.mpi diff --git a/src/allreduce_mock.h b/src/allreduce_mock.h index 96bc55800..1e08d57ed 100644 --- a/src/allreduce_mock.h +++ b/src/allreduce_mock.h @@ -53,6 +53,11 @@ class AllreduceMock : public AllreduceRobust { this->Verify(MockKey(rank, version_number, seq_counter, num_trial), "CheckPoint"); AllreduceRobust::CheckPoint(global_model, local_model); } + + virtual void LazyCheckPoint(const ISerializable *global_model) { + this->Verify(MockKey(rank, version_number, seq_counter, num_trial), "LazyCheckPoint"); + AllreduceRobust::LazyCheckPoint(global_model); + } private: // key to identify the mock stage