From 0a92c8574f60e7ba2f7af046de8e6926ad7b15d4 Mon Sep 17 00:00:00 2001 From: zmx Date: Fri, 12 Nov 2021 11:18:16 +0800 Subject: [PATCH] [heterps]fix ut for heter_pipeline_trainer.cc (#37136) * fix ut. test=develop * fix ut. test=develop --- paddle/fluid/framework/heter_pipeline_trainer_test.cc | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/paddle/fluid/framework/heter_pipeline_trainer_test.cc b/paddle/fluid/framework/heter_pipeline_trainer_test.cc index ef3d73b15a3..605578cab87 100644 --- a/paddle/fluid/framework/heter_pipeline_trainer_test.cc +++ b/paddle/fluid/framework/heter_pipeline_trainer_test.cc @@ -12,7 +12,8 @@ // See the License for the specific language governing permissions and // limitations under the License. -#if defined(PADDLE_WITH_PSCORE) +#if (defined(PADDLE_WITH_CUDA) || defined(PADDLE_WITH_HIP)) && \ + (defined PADDLE_WITH_PSCORE) #include "gtest/gtest.h" #include "paddle/fluid/framework/device_worker.h" #include "paddle/fluid/framework/device_worker_factory.h" -- GitLab