diff --git a/paddle/fluid/framework/communicator.h b/paddle/fluid/framework/communicator.h new file mode 100644 index 0000000000000000000000000000000000000000..ba8fb3e1731c73a12cded219df58b7b20bd11e16 --- /dev/null +++ b/paddle/fluid/framework/communicator.h @@ -0,0 +1,51 @@ +/* Copyright (c) 2016 PaddlePaddle Authors. All Rights Reserved. + +Licensed under the Apache License, Version 2.0 (the "License"); +you may not use this file except in compliance with the License. +You may obtain a copy of the License at + + http://www.apache.org/licenses/LICENSE-2.0 + +Unless required by applicable law or agreed to in writing, software +distributed under the License is distributed on an "AS IS" BASIS, +WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +See the License for the specific language governing permissions and +limitations under the License. */ + +#pragma once + +#include +#include +#include +#include +#include +#include "paddle/fluid/framework/data_layout.h" +#include "paddle/fluid/framework/ddim.h" +#include "paddle/fluid/framework/framework.pb.h" +#include "paddle/fluid/memory/memory.h" +#include "paddle/fluid/platform/device_context.h" +#include "paddle/fluid/platform/enforce.h" +#include "paddle/fluid/platform/place.h" + +namespace paddle { + +namespace framework { + +class Communicator { + public: + Communicator() {} + ~Communicator() {} + + // send grad + void send() {} + + void receive() {} + + void wait() {} + + private: + std::unique_ptr communicate_thread_; +}; + +} // namespace framework +} // namespace paddle