Fix unit test for send_and_recv_cpu & send_and_recv_gpu (PaddlePaddle…

…#37129)
huangxu96 · Nov 11, 2021 · a41447f · a41447f
1 parent 357425d
commit a41447f
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/paddle/fluid/operators/pscore/send_and_recv_op_cpu_test.cc b/paddle/fluid/operators/pscore/send_and_recv_op_cpu_test.cc
@@ -12,6 +12,7 @@ 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. */
 
+#if defined PADDLE_WITH_PSCORE
 #include <stdlib.h>
 #include <memory>
 #include <string>
@@ -272,3 +273,4 @@ TEST(SENDANDRECV, CPU) {
   server_thread.join();
   LOG(INFO) << "end server thread join";
 }
+#endif
diff --git a/paddle/fluid/operators/pscore/send_and_recv_op_gpu_test.cc b/paddle/fluid/operators/pscore/send_and_recv_op_gpu_test.cc
@@ -12,6 +12,9 @@ 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. */
 
+#if (defined PADDLE_WITH_CUDA || defined PADDLE_WITH_HIP) && \
+    (defined PADDLE_WITH_PSCORE)
+
 #include <stdlib.h>
 #include <memory>
 #include <string>
@@ -295,3 +298,4 @@ TEST(SENDANDRECV, GPU) {
   server_thread.join();
   LOG(INFO) << "end server thread join";
 }
+#endif