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

* fix ut. test=develop * fix ut. test=develop

[heterps]fix ut for heter_pipeline_trainer.cc (#37136)
* fix ut. test=develop * fix ut. test=develop
0a92c857 · zmx · GitHub · 36154ba9 · 0a92c857
隐藏空白更改
内联并排

Showing with 2 addition and 1 deletion

paddle/fluid/framework/heter_pipeline_trainer_test.cc paddle/fluid/framework/heter_pipeline_trainer_test.cc +2 -1

未找到文件。
--- 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"