未验证 提交 a41447f0 编写于 作者: Z zmx 提交者: GitHub

Fix unit test for send_and_recv_cpu & send_and_recv_gpu (#37129)

上级 357425d8
......@@ -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
......@@ -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
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册