[heterps]fix ut for heter_pipeline_trainer.cc (PaddlePaddle#37136)

* fix ut. test=develop * fix ut. test=develop
huangxu96 · Nov 12, 2021 · 0a92c85 · 0a92c85
1 parent 36154ba
commit 0a92c85
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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"