From bdfe87555f08fef5a4aff2976ce0e84d99730eab Mon Sep 17 00:00:00 2001 From: Thunderbrook Date: Sat, 8 May 2021 17:24:14 +0800 Subject: [PATCH] heter server test --- paddle/fluid/operators/pscore/heter_server_test.cc | 2 ++ 1 file changed, 2 insertions(+) diff --git a/paddle/fluid/operators/pscore/heter_server_test.cc b/paddle/fluid/operators/pscore/heter_server_test.cc index 1d072936f409c..df2eb70b144e4 100644 --- a/paddle/fluid/operators/pscore/heter_server_test.cc +++ b/paddle/fluid/operators/pscore/heter_server_test.cc @@ -20,6 +20,8 @@ limitations under the License. */ #include "gtest/gtest.h" #include "paddle/fluid/distributed/service/heter_client.h" #include "paddle/fluid/distributed/service/heter_server.h" +#include "paddle/fluid/framework/op_registry.h" + namespace framework = paddle::framework; namespace platform = paddle::platform; namespace distributed = paddle::distributed;